Spaces:

salomonsky
/

flux3

Running

vilarin commited on Aug 31

Commit

64c5e05

•

1 Parent(s): 8f6c590

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -36,7 +36,7 @@ if torch.cuda.is_available():
     pipe.to(device="cuda", dtype=torch.bfloat16)
 # Function
-@spaces.GPU(duration=120)
 def generate_image(
     prompt,
     width=768,

     pipe.to(device="cuda", dtype=torch.bfloat16)
 # Function
+@spaces.GPU()
 def generate_image(
     prompt,
     width=768,