Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -19,8 +19,7 @@ DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
|
19 |
pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev",
|
20 |
custom_pipeline="pipeline_flux_rf_inversion",
|
21 |
torch_dtype=torch.bfloat16)
|
22 |
-
|
23 |
-
pipe.fuse_lora(lora_scale=0.125)
|
24 |
#pipe.enable_lora()
|
25 |
pipe.to(DEVICE)
|
26 |
|
@@ -44,6 +43,7 @@ def check_style(stylezation):
|
|
44 |
|
45 |
def check_hyper_flux_lora(enable_hyper_flux):
|
46 |
if enable_hyper_flux:
|
|
|
47 |
pipe.fuse_lora(lora_scale=0.125)
|
48 |
return 8, 8, 4
|
49 |
else:
|
@@ -148,7 +148,7 @@ based on the implementations of [@raven38](https://github.com/raven38) & [@DarkM
|
|
148 |
minimum=0,
|
149 |
maximum=28,
|
150 |
step=1,
|
151 |
-
value=
|
152 |
)
|
153 |
eta = gr.Slider(
|
154 |
label="eta",
|
@@ -181,7 +181,7 @@ based on the implementations of [@raven38](https://github.com/raven38) & [@DarkM
|
|
181 |
minimum=1,
|
182 |
maximum=50,
|
183 |
step=1,
|
184 |
-
value=
|
185 |
)
|
186 |
eta_decay = gr.Checkbox(label="eta decay", value=False)
|
187 |
decay_power = gr.Slider(
|
@@ -206,7 +206,7 @@ based on the implementations of [@raven38](https://github.com/raven38) & [@DarkM
|
|
206 |
minimum=1,
|
207 |
maximum=50,
|
208 |
step=1,
|
209 |
-
value=
|
210 |
)
|
211 |
|
212 |
with gr.Row():
|
|
|
19 |
pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev",
|
20 |
custom_pipeline="pipeline_flux_rf_inversion",
|
21 |
torch_dtype=torch.bfloat16)
|
22 |
+
|
|
|
23 |
#pipe.enable_lora()
|
24 |
pipe.to(DEVICE)
|
25 |
|
|
|
43 |
|
44 |
def check_hyper_flux_lora(enable_hyper_flux):
|
45 |
if enable_hyper_flux:
|
46 |
+
pipe.load_lora_weights(hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-8steps-lora.safetensors"), lora_scale=0.125)
|
47 |
pipe.fuse_lora(lora_scale=0.125)
|
48 |
return 8, 8, 4
|
49 |
else:
|
|
|
148 |
minimum=0,
|
149 |
maximum=28,
|
150 |
step=1,
|
151 |
+
value=6,
|
152 |
)
|
153 |
eta = gr.Slider(
|
154 |
label="eta",
|
|
|
181 |
minimum=1,
|
182 |
maximum=50,
|
183 |
step=1,
|
184 |
+
value=28,
|
185 |
)
|
186 |
eta_decay = gr.Checkbox(label="eta decay", value=False)
|
187 |
decay_power = gr.Slider(
|
|
|
206 |
minimum=1,
|
207 |
maximum=50,
|
208 |
step=1,
|
209 |
+
value=28,
|
210 |
)
|
211 |
|
212 |
with gr.Row():
|