[Test] Align CosyVoice3 hardware mocks across platform layers
Signed-off-by: Sy03 <1370724210@qq.com>
This commit is contained in:
1 parent
6df57c5ba8
commit
7e6f4e1d8e
1 file changed
+3
@@ -589,7 +589,10 @@ class TestPipelineDiscovery:
|
||||
from vllm.platforms import current_platform
|
||||
from vllm.platforms.interface import DeviceCapability
|
||||
|
||||
from vllm_omni.platforms import current_omni_platform
|
||||
|
||||
monkeypatch.setattr(current_platform, "is_cuda", lambda: cuda)
|
||||
monkeypatch.setattr(current_omni_platform, "device_name", "cuda" if cuda else "cpu")
|
||||
capability = DeviceCapability(major, 0) if major is not None else None
|
||||
monkeypatch.setattr(current_platform, "get_device_capability", lambda: capability)
|
||||
monkeypatch.setattr(current_platform, "get_device_total_memory", lambda: memory_gib * 1024**3)
|
||||
|
||||
Reference in new issue
Block a user