Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 17:24:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

reference load cleanup

Browse Source 
Signed-off-by: vladmandic <mandic00@live.com>
This commit is contained in:
vladmandic
2025-10-25 23:25:10 -04:00
parent cffca11d5b
commit 512036d291
6 changed files with 28 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/ui_settings.py
+10 -2
View File
@@ -385,11 +385,19 @@ def create_quicksettings(interfaces):
def reference_submit(model):
if '@' not in model: # diffusers
loaded = modelloader.load_reference(model)
return model if loaded else shared.opts.sd_model_checkpoint
if loaded:
shared.opts.sd_model_checkpoint = model
sd_models.reload_model_weights(force=True)
return model
return shared.opts.sd_model_checkpoint
else: # civitai
model, url = model.split('@')
loaded = modelloader.load_civitai(model, url)
return loaded if loaded is not None else shared.opts.sd_model_checkpoint
if loaded is not None:
shared.opts.sd_model_checkpoint = loaded.title
sd_models.reload_model_weights(force=True)
return loaded
return shared.opts.sd_model_checkpoint
button_set_reference = gr.Button('Change reference', elem_id='change_reference', visible=False)
button_set_reference.click(