Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-20 01:31:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

Change cascade load order

Browse Source
This commit is contained in:
Disty0
2024-08-07 05:03:26 +03:00
parent 9818950991
commit 1cf87ef409
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_models.py
+2 -2
View File
@@ -1160,11 +1160,11 @@ def load_diffuser(checkpoint_info=None, already_loaded_state_dict=None, timer=No
if hasattr(sd_model, "set_progress_bar_config"):
sd_model.set_progress_bar_config(bar_format='Progress {rate_fmt}{postfix} {bar} {percentage:3.0f}% {n_fmt}/{total_fmt} {elapsed} {remaining}', ncols=80, colour='#327fba')
if model_type not in ['Stable Cascade']: # it will be handled in load_cascade if the detection works
sd_unet.load_unet(sd_model)
if "StableCascade" in sd_model.__class__.__name__: # detection can fail so we are applying post load here
from modules.model_stablecascade import cascade_post_load
cascade_post_load(sd_model)
if model_type not in ['Stable Cascade']: # it will be handled in load_cascade if the detection works
sd_unet.load_unet(sd_model)
timer.record("load")
if op == 'refiner':