K00B404 commited on
Commit
87b0e6b
·
verified ·
1 Parent(s): e40e5f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -43,7 +43,7 @@ transformer = FluxTransformer2DModel.from_single_file(
43
  torch_dtype=torch.bfloat16,
44
  )
45
  pipe = FluxPipeline.from_pretrained(
46
- "black-forest-labs/FLUX.1-schnell",
47
  transformer=transformer,
48
  torch_dtype=torch.bfloat16,
49
  )
@@ -88,7 +88,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
88
  width = gr.Slider(label="Width", minimum=256, maximum=1152, step=64, value=1024)
89
 
90
  with gr.Row():
91
- steps = gr.Slider(label="Inference Steps", minimum=6, maximum=25, step=1, value=8)
92
  scales = gr.Slider(label="Guidance Scale", minimum=0.0, maximum=5.0, step=0.1, value=3.5)
93
 
94
  seed = gr.Number(label="Seed (for reproducibility)", value=3413, precision=0)
 
43
  torch_dtype=torch.bfloat16,
44
  )
45
  pipe = FluxPipeline.from_pretrained(
46
+ "black-forest-labs/FLUX.1-dev",
47
  transformer=transformer,
48
  torch_dtype=torch.bfloat16,
49
  )
 
88
  width = gr.Slider(label="Width", minimum=256, maximum=1152, step=64, value=1024)
89
 
90
  with gr.Row():
91
+ steps = gr.Slider(label="Inference Steps", minimum=6, maximum=25, step=1, value=16)
92
  scales = gr.Slider(label="Guidance Scale", minimum=0.0, maximum=5.0, step=0.1, value=3.5)
93
 
94
  seed = gr.Number(label="Seed (for reproducibility)", value=3413, precision=0)