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

video tab alpha releasee

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-03-20 14:31:30 -04:00
parent 9a85e45cb4
commit 5c46904fd8
20 changed files with 874 additions and 172 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/cogvideo.py
+4 -4
View File
@@ -50,7 +50,7 @@ class Script(scripts.Script):
return [model, sampler, frames, guidance, offload, override, video_type, duration, loop, pad, interpolate, image, video]
def load(self, model):
if (shared.sd_model_type != 'cogvideox' or shared.sd_model.sd_model_checkpoint != model) and model != 'None':
if (shared.sd_model_type != 'cogvideo' or shared.sd_model.sd_model_checkpoint != model) and model != 'None':
sd_models.unload_model_weights('model')
shared.log.info(f'CogVideoX load: model="{model}"')
try:
@@ -64,7 +64,7 @@ class Script(scripts.Script):
shared.log.error(f'Load CogVideoX: {e}')
if debug:
errors.display(e, 'CogVideoX')
if shared.sd_model_type == 'cogvideox' and model != 'None':
if shared.sd_model_type == 'cogvideo' and model != 'None':
shared.sd_model.set_progress_bar_config(bar_format='Progress {rate_fmt}{postfix} {bar} {percentage:3.0f}% {n_fmt}/{total_fmt} {elapsed} {remaining} ' + '\x1b[38;5;71m', ncols=80, colour='#327fba')
shared.log.debug(f'CogVideoX load: class="{shared.sd_model.__class__.__name__}"')
if shared.sd_model is not None and model == 'None':
@@ -74,7 +74,7 @@ class Script(scripts.Script):
devices.torch_gc()
def offload(self, offload):
if shared.sd_model_type != 'cogvideox':
if shared.sd_model_type != 'cogvideo':
return
if offload == 'none':
sd_models.move_model(shared.sd_model, devices.device)
@@ -131,7 +131,7 @@ class Script(scripts.Script):
return img
def generate(self, p: processing.StableDiffusionProcessing, model: str):
if shared.sd_model_type != 'cogvideox':
if shared.sd_model_type != 'cogvideo':
return []
shared.log.info(f'CogVideoX: sampler={p.sampler} steps={p.steps} frames={p.frames} width={p.width} height={p.height} seed={p.seed} guidance={p.guidance}')
if p.sampler == 'DDIM':