Hugging Face's logo

Spaces:
cpuai
/
wan2-1-video-generation
Runtime error

App Files Community
multimodalart HF Staff commited on
Commit
c0724a8
·
verified ·
1 Parent(s): 481a175

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -141,7 +141,7 @@ pipeline = CausalInferencePipeline(
141
  vae=APP_STATE["current_vae_decoder"]
142
  )
143
 
144
- pipeline.to(gpu)
145
 
146
  # --- Additional Outputs Handler ---
147
  def handle_additional_outputs(status_html_update, video_update, webrtc_output):
 
141
  vae=APP_STATE["current_vae_decoder"]
142
  )
143
 
144
+ pipeline.to(dtype=torch.float16).to(gpu)
145
 
146
  # --- Additional Outputs Handler ---
147
  def handle_additional_outputs(status_html_update, video_update, webrtc_output):