Update forge_loader.py

This commit is contained in:
lllyasviel
2024-01-25 08:40:36 -08:00
parent 222f77538e
commit b088b9e807

View File

@@ -243,4 +243,7 @@ def load_model_for_a1111(timer, checkpoint_info=None, state_dict=None):
sd_model.vae_patcher_original = forge_object.vae.patcher
timer.record("get patcher")
# for token counter
sd_model.clip = sd_model.cond_stage_model
return sd_model