xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-05-15 15:33:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update tests/conftest.py

Browse Source 
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
Signed-off-by: Chukwuma Nwaugha <20521315+nwaughachukwuma@users.noreply.github.com>
This commit is contained in:
Chukwuma Nwaugha 2025-12-11 18:41:08 +00:00 committed by GitHub
parent 95af481818
commit dc763bbbe7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/conftest.py
View File

@ -265,7 +265,7 @@ class HfRunner:
if isinstance(x, dict): if isinstance(x, dict):
return {k: self.wrap_device(v, device) for k, v in x.items()} return {k: self.wrap_device(v, device) for k, v in x.items()}
if hasattr(x.device, "type") and x.device.type == device: if hasattr(x, "device") and x.device.type == device:
return x return x
return x.to(device) return x.to(device)