MCPcopy Create free account
hub / github.com/NVIDIA/TensorRT-LLM / device

Method device

tensorrt_llm/_torch/models/modeling_clip.py:224–225  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

222
223 @property
224 def device(self):
225 return get_parameter_device(self)
226
227 @torch.inference_mode()
228 def forward(self,

Callers 15

test_llama_sanityMethod · 0.45
test_llama_allclose_to_hfMethod · 0.45
test_clip_vision_allclose_to_hfMethod · 0.45
test_pixtral_vision_model_vs_hfFunction · 0.45
test_tensor_parallelismFunction · 0.45
test_siglip_vision_allclose_to_hfMethod · 0.45
test_bert_allclose_to_hfMethod · 0.45
test_llama_sanityMethod · 0.45
test_llama_allclose_to_hfMethod · 0.45
test_llama_verification_with_kv_cache_relocationMethod · 0.45
test_mixtral_sanityMethod · 0.45
test_mixtral_allclose_to_hfMethod · 0.45

Calls

no outgoing calls

Tested by 15

test_llama_sanityMethod · 0.36
test_llama_allclose_to_hfMethod · 0.36
test_clip_vision_allclose_to_hfMethod · 0.36
test_pixtral_vision_model_vs_hfFunction · 0.36
test_tensor_parallelismFunction · 0.36
test_siglip_vision_allclose_to_hfMethod · 0.36
test_bert_allclose_to_hfMethod · 0.36
test_llama_sanityMethod · 0.36
test_llama_allclose_to_hfMethod · 0.36
test_llama_verification_with_kv_cache_relocationMethod · 0.36
test_mixtral_sanityMethod · 0.36
test_mixtral_allclose_to_hfMethod · 0.36