lnyan commited on
Commit
5cea7d2
·
1 Parent(s): 6ce893a
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -205,7 +205,7 @@ class FluxGenerator:
205
  else:
206
  return None, str(opts.seed), None, "Your generated image may contain NSFW content."
207
 
208
- @spaces.GPU(duration=200)
209
  def create_demo(model_name: str, device: str = "cuda", offload: bool = False):
210
  generator = FluxGenerator(model_name, device, offload)
211
  is_schnell = model_name == "flux-schnell"
 
205
  else:
206
  return None, str(opts.seed), None, "Your generated image may contain NSFW content."
207
 
208
+ @spaces.GPU(duration=180)
209
  def create_demo(model_name: str, device: str = "cuda", offload: bool = False):
210
  generator = FluxGenerator(model_name, device, offload)
211
  is_schnell = model_name == "flux-schnell"