biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

[Misc] rename torch_dtype to dtype (#26695)

Browse Source 
Signed-off-by: wangxiyuan <wangxiyuan1007@gmail.com>
This commit is contained in:
wangxiyuan
2025-10-15 20:11:48 +08:00
committed by GitHub
parent f93e348010
commit 8f4b313c37
30 changed files with 52 additions and 55 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/multimodal/pooling/test_intern_vit.py
View File

@@ -38,7 +38,7 @@ def run_intern_vit_test(
config.norm_type = "rms_norm"
hf_model = AutoModel.from_pretrained(
model, torch_dtype=torch_dtype, trust_remote_code=True
model, dtype=torch_dtype, trust_remote_code=True
).to("cuda")
hf_outputs_per_image = [
hf_model(pixel_value.to("cuda")).last_hidden_state

2
tests/models/multimodal/pooling/test_radio.py
View File

@@ -45,7 +45,7 @@ def run_radio_test(
hf_model = AutoModel.from_pretrained(
model_id,
config=config,
torch_dtype=torch_dtype,
dtype=torch_dtype,
trust_remote_code=True,
).to("cuda")
hf_model.eval()