jonluca commited on
Commit
942edfa
Β·
verified Β·
1 Parent(s): 41a9d22

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -89,15 +89,15 @@ def generate_video_gpu(
89
  width, height = int(width), int(height)
90
  negative_prompt = "" # not applicable in the inference
91
  print("Predicting video...")
 
92
  frames: List[PIL.Image.Image] = model(
93
  prompt=prompt,
94
  height=height,
95
  width=width,
96
  num_frames=video_length,
97
- seed=seed,
98
  num_inference_steps=num_inference_steps,
99
- guidance_scale=guidance_scale,
100
- num_videos_per_prompt=1
101
  ).frames[0]
102
 
103
  output_video = export_to_video(frames, fps=15)
 
89
  width, height = int(width), int(height)
90
  negative_prompt = "" # not applicable in the inference
91
  print("Predicting video...")
92
+ generator = torch.manual_seed(seed)
93
  frames: List[PIL.Image.Image] = model(
94
  prompt=prompt,
95
  height=height,
96
  width=width,
97
  num_frames=video_length,
98
+ generator=generator,
99
  num_inference_steps=num_inference_steps,
100
+ guidance_scale=guidance_scale
 
101
  ).frames[0]
102
 
103
  output_video = export_to_video(frames, fps=15)