Skip to content

Commit c3cf18e

Browse files
tests(mm): attempt to fix windows model id tests
1 parent 40601d1 commit c3cf18e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/model_identification/stripped_model_on_disk.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,7 @@ def dress(cls, v: Any):
6767
)
6868
case {"shape": shape, "dtype": dtype_str, "fakeTensor": True}:
6969
dtype = cls.STR_TO_DTYPE[dtype_str]
70-
return torch.empty(shape, dtype=dtype)
70+
return torch.empty(shape, dtype=dtype, device="meta")
7171
case dict():
7272
return {k: cls.dress(v) for k, v in v.items()}
7373
case list() | tuple():

0 commit comments

Comments
 (0)