From ffdd69117ea46109814bfebbac7b89136b023283 Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Sat, 30 Aug 2025 09:13:43 -0400 Subject: [PATCH] fix skyreels Signed-off-by: Vladimir Mandic --- modules/video_models/models_def.py | 10 +++++----- modules/video_models/video_overrides.py | 2 ++ 2 files changed, 7 insertions(+), 5 deletions(-) diff --git a/modules/video_models/models_def.py b/modules/video_models/models_def.py index 173449874..6ee8565c5 100644 --- a/modules/video_models/models_def.py +++ b/modules/video_models/models_def.py @@ -212,14 +212,14 @@ models = { ], 'SkyReels V2': [ Model(name='None'), - Model(name='SkyReels-V2 T2I-DF 1.3B-540P', + Model(name='SkyReels-V2 T2V-DF 1.3B-540P', url='https://huggingface.co/Skywork/SkyReels-V2-DF-1.3B-540P-Diffusers', repo='Skywork/SkyReels-V2-DF-1.3B-540P-Diffusers', repo_cls=diffusers.SkyReelsV2DiffusionForcingPipeline, repo_revision='refs/pr/1', te_cls=transformers.UMT5EncoderModel, dit_cls=diffusers.SkyReelsV2Transformer3DModel), - Model(name='SkyReels-V2 T2I-DF 14B-720P', + Model(name='SkyReels-V2 T2V-DF 14B-720P', url='https://huggingface.co/Skywork/SkyReels-V2-DF-14B-720P-Diffusers', repo='Skywork/SkyReels-V2-DF-14B-720P-Diffusers', repo_cls=diffusers.SkyReelsV2DiffusionForcingPipeline, @@ -233,7 +233,7 @@ models = { repo_revision='refs/pr/1', te_cls=transformers.UMT5EncoderModel, dit_cls=diffusers.SkyReelsV2Transformer3DModel), - Model(name='SkyReels-V2 T2I 14B-720P', + Model(name='SkyReels-V2 T2V 14B-720P', url='https://huggingface.co/Skywork/SkyReels-V2-T2V-14B-720P-Diffusers', repo='Skywork/SkyReels-V2-T2V-14B-720P-Diffusers', repo_cls=diffusers.SkyReelsV2Pipeline, @@ -321,13 +321,13 @@ models = { dit_cls=diffusers.CogVideoXTransformer3DModel), ], 'nVidia Cosmos': [ - Model(name='nvidia Cosmos Predict2 2B I2I', + Model(name='nvidia Cosmos Predict2 2B I2V', url='https://huggingface.co/nvidia/Cosmos-Predict2-2B-Text2Image', repo='nvidia/Cosmos-Predict2-2B-Video2World', repo_cls=diffusers.Cosmos2VideoToWorldPipeline, te_cls=transformers.T5EncoderModel, dit_cls=diffusers.CosmosTransformer3DModel), - Model(name='nvidia Cosmos Predict2 2B I2I', + Model(name='nvidia Cosmos Predict2 2B I2V', url='https://huggingface.co/nvidia/Cosmos-Predict2-2B-Text2Image', repo='nvidia/Cosmos-Predict2-2B-Video2World', repo_cls=diffusers.Cosmos2VideoToWorldPipeline, diff --git a/modules/video_models/video_overrides.py b/modules/video_models/video_overrides.py index 07cf1cd23..5ce02e20a 100644 --- a/modules/video_models/video_overrides.py +++ b/modules/video_models/video_overrides.py @@ -46,3 +46,5 @@ def set_overrides(p: processing.StableDiffusionProcessingVideo, selected: Model) if 'LTX' in cls: p.task_args['width'] = 32 * (p.width // 32) p.task_args['height'] = 32 * (p.height // 32) + if 'SkyReelsV2DiffusionForcing' in cls: + p.task_args['overlap_history'] = 17