Nymbo commited on
Commit
3e4c5c9
·
verified ·
1 Parent(s): 1b388ba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,7 +7,7 @@ from huggingface_hub import snapshot_download
7
  from pyramid_dit import PyramidDiTForVideoGeneration
8
  from diffusers.utils import export_to_video
9
 
10
- import spaces
11
  import uuid
12
 
13
  is_canonical = True if os.environ.get("SPACE_ID") == "Pyramid-Flow/pyramid-flow" else False
@@ -61,7 +61,7 @@ def load_model():
61
  model = load_model()
62
 
63
  # Text-to-video generation function
64
- @spaces.GPU(duration=140)
65
  def generate_video(prompt, image=None, duration=3, guidance_scale=9, video_guidance_scale=5, frames_per_second=8, progress=gr.Progress(track_tqdm=True)):
66
  multiplier = 1.2 if is_canonical else 3.0
67
  temp = int(duration * multiplier) + 1
 
7
  from pyramid_dit import PyramidDiTForVideoGeneration
8
  from diffusers.utils import export_to_video
9
 
10
+ # import spaces
11
  import uuid
12
 
13
  is_canonical = True if os.environ.get("SPACE_ID") == "Pyramid-Flow/pyramid-flow" else False
 
61
  model = load_model()
62
 
63
  # Text-to-video generation function
64
+ # @spaces.GPU(duration=140)
65
  def generate_video(prompt, image=None, duration=3, guidance_scale=9, video_guidance_scale=5, frames_per_second=8, progress=gr.Progress(track_tqdm=True)):
66
  multiplier = 1.2 if is_canonical else 3.0
67
  temp = int(duration * multiplier) + 1