fix and rework #11113

This commit is contained in:
AUTOMATIC 2023-06-27 09:26:18 +03:00
parent a96687682a
commit d06af4e517

View File

@ -522,9 +522,9 @@ class Api:
return options return options
def set_config(self, req: Dict[str, Any]): def set_config(self, req: Dict[str, Any]):
checkpoint_key="sd_model_checkpoint" checkpoint_name = req.get("sd_model_checkpoint", None)
if checkpoint_key in req and str(req[checkpoint_key]) not in checkpoint_alisases: if checkpoint_name is not None and checkpoint_name not in checkpoint_alisases:
raise RuntimeError(f"model {v!r} not found") raise RuntimeError(f"model {checkpoint_name!r} not found")
for k, v in req.items(): for k, v in req.items():
shared.opts.set(k, v) shared.opts.set(k, v)