Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 |
-
|
| 268 |
-
|
| 269 |
-
|
| 270 |
-
|
| 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)
|