Allow passing in key on register

2024-08-30 20:32:17 +00:00 · 2024-02-23 12:57:54 -05:00
parent 80ad14d89f
commit 09295ae43b
2 changed files with 6 additions and 4 deletions
--- a/invokeai/backend/model_manager/probe.py
+++ b/invokeai/backend/model_manager/probe.py
@ -188,7 +188,7 @@ class ModelProbe(object):
                and fields["prediction_type"] == SchedulerPredictionType.VPrediction
            )

-        model_info = ModelConfigFactory.make_config(fields)
+        model_info = ModelConfigFactory.make_config(fields, key=fields.get("key", None))
        return model_info

    @classmethod