Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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-
|
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=
|
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)
|