lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-03-14 01:19:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

model load entry for all model loads from webui

Browse Source 
including startup, checkpoint selection, and that HTML tab of some preview images
This commit is contained in:
layerdiffusion
2024-08-06 14:38:40 -07:00
parent 6d789653b9
commit b4ca5d7420
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
modules_forge/main_entry.py
View File

@@ -40,12 +40,17 @@ def make_checkpoint_manager_ui():
return
def model_load_entry():
sd_models.load_model()
return
def checkpoint_change(ckpt_name):
print(f'Checkpoint Selected: {ckpt_name}')
shared.opts.set('sd_model_checkpoint', ckpt_name)
shared.opts.save(shared.config_filename)
sd_models.load_model()
model_load_entry()
return
@@ -68,5 +73,5 @@ def forge_main_entry():
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)
main_thread.async_run(model_load_entry)
return