Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -515,6 +515,7 @@ def _generate(
|
|
| 515 |
width=width,
|
| 516 |
num_frames=num_frames,
|
| 517 |
num_inference_steps=requested_steps,
|
|
|
|
| 518 |
generator=torch.Generator("cpu").manual_seed(int(seed)),
|
| 519 |
)
|
| 520 |
|
|
|
|
| 515 |
width=width,
|
| 516 |
num_frames=num_frames,
|
| 517 |
num_inference_steps=requested_steps,
|
| 518 |
+
output_type="pt",
|
| 519 |
generator=torch.Generator("cpu").manual_seed(int(seed)),
|
| 520 |
)
|
| 521 |
|