Maximofn commited on
Commit
7265fcb
·
1 Parent(s): 264ba49

feat(src): :rocket: Add Zero GPU decorator

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -7,8 +7,9 @@ from datetime import datetime
7
  import gradio as gr
8
 
9
  from hyvideo.config import parse_args
 
10
 
11
-
12
  def initialize_model(model):
13
  quant_config = BitsAndBytesConfig(load_in_8bit=True)
14
 
@@ -30,6 +31,7 @@ def initialize_model(model):
30
 
31
  return pipeline
32
 
 
33
  def generate_video(
34
  pipeline,
35
  prompt,
 
7
  import gradio as gr
8
 
9
  from hyvideo.config import parse_args
10
+ import spaces
11
 
12
+ @spaces.GPU
13
  def initialize_model(model):
14
  quant_config = BitsAndBytesConfig(load_in_8bit=True)
15
 
 
31
 
32
  return pipeline
33
 
34
+ @spaces.GPU
35
  def generate_video(
36
  pipeline,
37
  prompt,