evalstate HF Staff commited on
Commit
ec1ea11
·
verified ·
1 Parent(s): 502fa54

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -13,7 +13,7 @@ pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", tor
13
  MAX_SEED = np.iinfo(np.int32).max
14
  MAX_IMAGE_SIZE = 2048
15
 
16
- @spaces.GPU(duration=15)
17
  def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=4, progress=gr.Progress(track_tqdm=True)):
18
  """
19
  Generate an image using the Flux 1 Schnell Image Generator.
@@ -102,7 +102,7 @@ with gr.Blocks(css=css) as demo:
102
  num_inference_steps = gr.Slider(
103
  label="Number of inference steps",
104
  minimum=1,
105
- maximum=50,
106
  step=1,
107
  value=4,
108
  )
 
13
  MAX_SEED = np.iinfo(np.int32).max
14
  MAX_IMAGE_SIZE = 2048
15
 
16
+ @spaces.GPU(duration=10)
17
  def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=4, progress=gr.Progress(track_tqdm=True)):
18
  """
19
  Generate an image using the Flux 1 Schnell Image Generator.
 
102
  num_inference_steps = gr.Slider(
103
  label="Number of inference steps",
104
  minimum=1,
105
+ maximum=16,
106
  step=1,
107
  value=4,
108
  )