linoyts HF Staff commited on
Commit
dad60fc
Β·
verified Β·
1 Parent(s): 5b1caa6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -181,7 +181,7 @@ def process_video_for_control(video, control_type):
181
  else:
182
  return video
183
 
184
- @spaces.GPU(duration=120)
185
  def generate_video(
186
  reference_video,
187
  prompt,
@@ -291,7 +291,7 @@ def generate_video(
291
  image_cond_noise_scale=image_cond_noise_scale,
292
  guidance_scale=guidance_scale,
293
  guidance_rescale=guidance_rescale,
294
- generator=torch.Generator().manual_seed(seed),
295
  output_type="pil",
296
  ).frames[0]
297
 
 
181
  else:
182
  return video
183
 
184
+ @spaces.GPU(duration=160)
185
  def generate_video(
186
  reference_video,
187
  prompt,
 
291
  image_cond_noise_scale=image_cond_noise_scale,
292
  guidance_scale=guidance_scale,
293
  guidance_rescale=guidance_rescale,
294
+ generator=torch.Generator(device="cuda").manual_seed(seed),
295
  output_type="pil",
296
  ).frames[0]
297