fix: ModelProbe.probe needs to return a size field

Kevin Turner · Kevin Turner · commit 5ebccb3c93cf · 2025-04-04T14:47:15.000-07:00
diff --git a/invokeai/backend/model_manager/legacy_probe.py b/invokeai/backend/model_manager/legacy_probe.py
@@ -207,6 +207,7 @@ def probe(
         )
         fields["format"] = ModelFormat(fields.get("format")) if "format" in fields else probe.get_format()
         fields["hash"] = fields.get("hash") or ModelHash(algorithm=hash_algo).hash(model_path)
+        fields["size"] = fields.get("size") or -1  # FIXME: placeholder to narrow down tests
 
         fields["default_settings"] = fields.get("default_settings")
 

Original file line number	Diff line number	Diff line change
`@@ -207,6 +207,7 @@ def probe(`
`207`	`207`	`)`
`208`	`208`	`fields["format"] = ModelFormat(fields.get("format")) if "format" in fields else probe.get_format()`
`209`	`209`	`fields["hash"] = fields.get("hash") or ModelHash(algorithm=hash_algo).hash(model_path)`
	`210`	`+ fields["size"] = fields.get("size") or -1 # FIXME: placeholder to narrow down tests`
`210`	`211`
`211`	`212`	`fields["default_settings"] = fields.get("default_settings")`
`212`	`213`