rahul7star commited on
Commit
5f3e179
·
verified ·
1 Parent(s): 1d2fe7b

Update app_t2v.py

Browse files
Files changed (1) hide show
  1. app_t2v.py +1 -5
app_t2v.py CHANGED
@@ -20,12 +20,8 @@ pipe = WanPipeline.from_pretrained(
20
  pipe.to(device)
21
 
22
  # Core inference function
23
- def get_duration(
24
- prompt, negative_prompt, height, width, num_frames, guidance_scale, guidance_scale_2, num_steps
25
- ):
26
- return steps * 15
27
 
28
- @spaces.GPU(duration=get_duration)
29
  def generate_video(prompt, negative_prompt, height, width, num_frames, guidance_scale, guidance_scale_2, num_steps):
30
  with torch.autocast("cuda", dtype=dtype):
31
  output = pipe(
 
20
  pipe.to(device)
21
 
22
  # Core inference function
 
 
 
 
23
 
24
+ @spaces.GPU(duration=200)
25
  def generate_video(prompt, negative_prompt, height, width, num_frames, guidance_scale, guidance_scale_2, num_steps):
26
  with torch.autocast("cuda", dtype=dtype):
27
  output = pipe(