dagloop5 commited on
Commit
f6f9a22
·
verified ·
1 Parent(s): 9d53740

Update app.py

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