tests(mm): attempt to fix windows model id tests

2026-04-23 03:00:31 -04:00 · 2025-10-10 18:24:10 +11:00
parent 40601d1f67
commit c3cf18eaf8
1 changed files with 1 additions and 1 deletions
--- a/tests/model_identification/stripped_model_on_disk.py
+++ b/tests/model_identification/stripped_model_on_disk.py
@@ -67,7 +67,7 @@ class StrippedModelOnDisk(ModelOnDisk):
                )
            case {"shape": shape, "dtype": dtype_str, "fakeTensor": True}:
                dtype = cls.STR_TO_DTYPE[dtype_str]
-                return torch.empty(shape, dtype=dtype)
+                return torch.empty(shape, dtype=dtype, device="meta")
            case dict():
                return {k: cls.dress(v) for k, v in v.items()}
            case list() | tuple():