Skip to content

Commit 6c8afb2

Browse files
committed
fix merge missing
1 parent aa6a8ac commit 6c8afb2

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

modules/api/api.py

+4-3
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@
2424
from modules.textual_inversion.textual_inversion import create_embedding, train_embedding
2525
from modules.hypernetworks.hypernetwork import create_hypernetwork, train_hypernetwork
2626
from PIL import PngImagePlugin, Image
27+
from modules.sd_models import unload_model_weights, reload_model_weights, checkpoint_aliases
2728
from modules.sd_models_config import find_checkpoint_config_near_filename
2829
from modules.realesrgan_model import get_realesrgan_models
2930
from modules import devices
@@ -704,12 +705,12 @@ def interruptapi(self):
704705
return {}
705706

706707
def unloadapi(self):
707-
sd_models.unload_model_weights()
708+
unload_model_weights()
708709

709710
return {}
710711

711712
def reloadapi(self):
712-
sd_models.send_model_to_device(shared.sd_model)
713+
reload_model_weights()
713714

714715
return {}
715716

@@ -729,7 +730,7 @@ def get_config(self):
729730

730731
def set_config(self, req: dict[str, Any]):
731732
checkpoint_name = req.get("sd_model_checkpoint", None)
732-
if checkpoint_name is not None and checkpoint_name not in sd_models.checkpoint_aliases:
733+
if checkpoint_name is not None and checkpoint_name not in checkpoint_aliases:
733734
raise RuntimeError(f"model {checkpoint_name!r} not found")
734735

735736
for k, v in req.items():

0 commit comments

Comments
 (0)