p1atdev commited on
Commit
dae5e79
1 Parent(s): 3513d23

chore: set duration

Browse files
Files changed (1) hide show
  1. diffusion.py +2 -2
diffusion.py CHANGED
@@ -94,10 +94,10 @@ def image_generation_config_ui():
94
  class ImageGenerator:
95
  # pipe: StableDiffusionXLPipeline
96
 
97
- def __init__(self, model_name: str = "cagliostrolab/animagine-xl-3.1"):
98
  pass
99
 
100
- @spaces.GPU()
101
  def generate(
102
  self,
103
  prompt: str,
 
94
  class ImageGenerator:
95
  # pipe: StableDiffusionXLPipeline
96
 
97
+ def __init__(self):
98
  pass
99
 
100
+ @spaces.GPU(duration=30)
101
  def generate(
102
  self,
103
  prompt: str,