Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -174,7 +174,8 @@ def generate_video(prompt: str):
|
|
174 |
#model = manager.load_model(WAN_MODEL_PATH)
|
175 |
|
176 |
# Set up pipeline
|
177 |
-
pipeline = WanVideoPusaPipeline(model=model_manager)
|
|
|
178 |
#pipeline.set_lora_adapters(LORA_PATH)
|
179 |
|
180 |
# Generate video
|
|
|
174 |
#model = manager.load_model(WAN_MODEL_PATH)
|
175 |
|
176 |
# Set up pipeline
|
177 |
+
#pipeline = WanVideoPusaPipeline(model=model_manager)
|
178 |
+
pipeline = WanVideoPusaPipeline.from_model_manager(model_manager, torch_dtype=torch.bfloat16, device="cuda")
|
179 |
#pipeline.set_lora_adapters(LORA_PATH)
|
180 |
|
181 |
# Generate video
|