add nvidia-sana-video-2b

Signed-off-by: Vladimir Mandic <mandic00@live.com>

modules/video_models/models_def.py

@@ -367,6 +367,14 @@ try:
                 te_cls=getattr(transformers, 'T5EncoderModel', None),
                 dit_cls=getattr(diffusers, 'CosmosTransformer3DModel', None)),
         ],
+        'nVidia SANA': [
+            Model(name='SANA Video 2B 480p T2V',
+                url='https://huggingface.co/Efficient-Large-Model/SANA-Video_2B_480p_diffusers',
+                repo='Efficient-Large-Model/SANA-Video_2B_480p_diffusers',
+                repo_cls=getattr(diffusers, 'SanaVideoPipeline', None),
+                te_cls=getattr(transformers, 'Gemma2Model', None),
+                dit_cls=getattr(diffusers, 'SanaVideoTransformer3DModel', None)),
+        ],
         'Kandinsky': [
             Model(name='Kandinsky 5.0 Lite 5s SFT T2V',
                 url='https://huggingface.co/ai-forever/Kandinsky-5.0-T2V-Lite-sft-5s-Diffusers',

modules/video_models/video_load.py

@@ -1,6 +1,8 @@
 import os
 import copy
 import time
+import transformers # pylint: disable=unused-import
+import diffusers
 from modules import shared, errors, sd_models, sd_checkpoint, model_quant, devices, sd_hijack_te, sd_hijack_vae
 from modules.video_models import models_def, video_utils, video_overrides, video_cache
 
@@ -174,7 +176,6 @@ def load_upscale_vae():
         shared.log.warning('Video decode: upscale VAE unsupported')
         return
 
-    import diffusers
     repo_id = 'spacepxl/Wan2.1-VAE-upscale2x'
     subfolder = "diffusers/Wan2.1_VAE_upscale2x_imageonly_real_v1"
     vae_decode = diffusers.AutoencoderKLWan.from_pretrained(repo_id, subfolder=subfolder, cache_dir=shared.opts.hfcache_dir)