Commit 8ca34ad6 authored by Su Wei's avatar Su Wei

add model exists status check to modeuls/api/api.py , /sdapi/v1/options [POST]

parent baf6946e
...@@ -22,7 +22,7 @@ from modules.textual_inversion.textual_inversion import create_embedding, train_ ...@@ -22,7 +22,7 @@ from modules.textual_inversion.textual_inversion import create_embedding, train_
from modules.textual_inversion.preprocess import preprocess from modules.textual_inversion.preprocess import preprocess
from modules.hypernetworks.hypernetwork import create_hypernetwork, train_hypernetwork from modules.hypernetworks.hypernetwork import create_hypernetwork, train_hypernetwork
from PIL import PngImagePlugin,Image from PIL import PngImagePlugin,Image
from modules.sd_models import checkpoints_list, unload_model_weights, reload_model_weights from modules.sd_models import checkpoints_list, unload_model_weights, reload_model_weights,checkpoint_alisases
from modules.sd_models_config import find_checkpoint_config_near_filename from modules.sd_models_config import find_checkpoint_config_near_filename
from modules.realesrgan_model import get_realesrgan_models from modules.realesrgan_model import get_realesrgan_models
from modules import devices from modules import devices
...@@ -515,6 +515,11 @@ class Api: ...@@ -515,6 +515,11 @@ class Api:
def set_config(self, req: Dict[str, Any]): def set_config(self, req: Dict[str, Any]):
for k, v in req.items(): for k, v in req.items():
if k == "sd_model_checkpoint":
checkpoint_info = checkpoint_alisases.get(v, None)
if checkpoint_info is None:
print(f"model [{v}] not founded, skip config saving process")
return
shared.opts.set(k, v) shared.opts.set(k, v)
shared.opts.save(shared.config_filename) shared.opts.save(shared.config_filename)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment