linoyts HF Staff commited on
Commit
498af35
Β·
verified Β·
1 Parent(s): 2af7694

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -136,7 +136,7 @@ def process_video_for_control(video, control_type):
136
  else:
137
  return video
138
 
139
- @spaces.GPU(duration=120)
140
  def generate_video(
141
  reference_video,
142
  prompt,
@@ -246,7 +246,7 @@ def generate_video(
246
  image_cond_noise_scale=image_cond_noise_scale,
247
  guidance_scale=guidance_scale,
248
  guidance_rescale=guidance_rescale,
249
- generator=torch.Generator().manual_seed(seed),
250
  output_type="pil",
251
  ).frames[0]
252
 
 
136
  else:
137
  return video
138
 
139
+ @spaces.GPU(duration=160)
140
  def generate_video(
141
  reference_video,
142
  prompt,
 
246
  image_cond_noise_scale=image_cond_noise_scale,
247
  guidance_scale=guidance_scale,
248
  guidance_rescale=guidance_rescale,
249
+ generator=torch.Generator(device="cuda").manual_seed(seed),
250
  output_type="pil",
251
  ).frames[0]
252