multimodalart HF Staff commited on
Commit
35ce3f1
·
verified ·
1 Parent(s): 8d82dd3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -264,10 +264,10 @@ model_loaders = [
264
  MODELS_AND_NODES["model_high_noise"], # This is the other UNET + LoRA
265
  MODELS_AND_NODES["clip_vision"],
266
  ]
267
- #model_management.load_models_gpu([
268
- # loader[0].patcher if hasattr(loader[0], 'patcher') else loader[0] for loader in model_loaders
269
- #])
270
- #print("All models loaded successfully!")
271
 
272
  # --- Main Video Generation Logic ---
273
  @spaces.GPU(duration=120)
 
264
  MODELS_AND_NODES["model_high_noise"], # This is the other UNET + LoRA
265
  MODELS_AND_NODES["clip_vision"],
266
  ]
267
+ model_management.load_models_gpu([
268
+ loader[0].patcher if hasattr(loader[0], 'patcher') else loader[0] for loader in model_loaders
269
+ ])
270
+ print("All models loaded successfully!")
271
 
272
  # --- Main Video Generation Logic ---
273
  @spaces.GPU(duration=120)