daoyuan98 commited on
Commit
4bc8a6f
1 Parent(s): c8690a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -256,7 +256,7 @@ MAX_IMAGE_SIZE = 1024
256
 
257
  pipe.flux_pipe_call_that_returns_an_iterable_of_images = flux_pipe_call_that_returns_an_iterable_of_images.__get__(pipe)
258
 
259
- @spaces.GPU(duration=75)
260
  def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidance_scale=3.5, num_inference_steps=28, progress=gr.Progress(track_tqdm=True)):
261
  torch.cuda.empty_cache()
262
  if randomize_seed:
 
256
 
257
  pipe.flux_pipe_call_that_returns_an_iterable_of_images = flux_pipe_call_that_returns_an_iterable_of_images.__get__(pipe)
258
 
259
+ @spaces.GPU(duration=30)
260
  def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidance_scale=3.5, num_inference_steps=28, progress=gr.Progress(track_tqdm=True)):
261
  torch.cuda.empty_cache()
262
  if randomize_seed: