github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

Disable device meta for spandrel

Browse Source
This commit is contained in:
Billy
2025-03-17 11:30:05 +11:00
parent 654e992630
commit 7d5687e9ff
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/strip_models.py
View File

@@ -42,7 +42,7 @@ def dress(v):
match v:
case {"shape": shape, "dtype": dtype_str, "fakeTensor": True}:
dtype = STR_TO_DTYPE[dtype_str]
return torch.empty(shape, dtype=dtype, device="meta")
return torch.empty(shape, dtype=dtype)
case dict():
return {k: dress(v) for k, v in v.items()}
case list() | tuple():
@@ -54,7 +54,7 @@ def dress(v):
def load_stripped_model(path: Path, *args, **kwargs):
with open(path, "r") as f:
contents = json.load(f)
return { "state_dict": dress(contents) }
return dress(contents)
def create_stripped_model(original_model_path: Path, stripped_model_path: Path) -> ModelOnDisk: