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

Merge branch 'conftest/generate_beam_search/simplify-return-value' of github.com:nwaughachukwuma/vllm into conftest/generate_beam_search/simplify-return-value

Browse Source
This commit is contained in:
Chukwuma Nwaugha 2025-12-11 19:39:21 +00:00
commit b8afaf9c18
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)