Hugging Face's logo

Spaces:
huan2hoang3
/
flux-lab-light
Sleeping

App Files Community
vilarin commited on
Commit
64c5e05
·
verified ·
1 Parent(s): 8f6c590

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -36,7 +36,7 @@ if torch.cuda.is_available():
36
  pipe.to(device="cuda", dtype=torch.bfloat16)
37
 
38
  # Function
39
- @spaces.GPU(duration=120)
40
  def generate_image(
41
  prompt,
42
  width=768,
 
36
  pipe.to(device="cuda", dtype=torch.bfloat16)
37
 
38
  # Function
39
+ @spaces.GPU()
40
  def generate_image(
41
  prompt,
42
  width=768,