Spaces:
Runtime error
Runtime error
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():
|