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

fix skyreels

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-08-30 09:13:43 -04:00
parent 1dc22650c2
commit ffdd69117e
2 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/video_models/models_def.py
+5 -5
View File
@@ -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,
modules/video_models/video_overrides.py
+2
View File
@@ -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