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

pipeline load with variant fallback

Browse Source
This commit is contained in:
Vladimir Mandic
2024-03-14 15:09:05 -04:00
parent e971e4fbc0
commit 20bc82a4ce
3 changed files with 48 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/stablevideodiffusion.py
+1 -2
View File
@@ -71,11 +71,10 @@ class Script(scripts.Script):
shared.log.error(f'SVD: no checkpoint for {model_name}')
modelloader.load_reference(model_path, variant='fp16')
c = shared.sd_model.__class__.__name__
model_loaded = shared.sd_model.sd_checkpoint_info.model_name
model_loaded = shared.sd_model.sd_checkpoint_info.model_name if shared.sd_model is not None else None
if model_name != model_loaded or c != 'StableVideoDiffusionPipeline':
shared.opts.sd_model_checkpoint = model_path
sd_models.reload_model_weights()
model_loaded = sd_models.model_data.sd_model.sd_checkpoint_info
# set params
if override_resolution: