rahul7star commited on
Commit
afa05d9
·
verified ·
1 Parent(s): 60daa3f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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