better model load logic

This commit is contained in:
layerdiffusion
2024-08-06 14:34:57 -07:00
parent c1b23bd494
commit 6d789653b9
3 changed files with 4 additions and 20 deletions

View File

@@ -66,4 +66,7 @@ def forge_main_entry():
ui_checkpoint.change(lambda x: main_thread.async_run(checkpoint_change, x), inputs=[ui_checkpoint], show_progress=False)
ui_vae.change(lambda x: main_thread.async_run(vae_change, x), inputs=[ui_vae], show_progress=False)
ui_clip_skip.change(lambda x: main_thread.async_run(clip_skip_change, x), inputs=[ui_clip_skip], show_progress=False)
# Load Model
main_thread.async_run(sd_models.load_model)
return