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