Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -19,8 +19,8 @@ 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 |
-
|
24 |
pipe.to(DEVICE)
|
25 |
|
26 |
examples = [[Image.open("cat.jpg"), "a tiger", 0, 0.38, 0.5, 0.9, 28, 28, 789385745, False]]
|
|
|
19 |
pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev",
|
20 |
custom_pipeline="pipeline_flux_rf_inversion",
|
21 |
torch_dtype=torch.bfloat16)
|
22 |
+
pipe.load_lora_weights(hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-8steps-lora.safetensors"), lora_scale=0.125)
|
23 |
+
pipe.fuse_lora(lora_scale=0.125)
|
24 |
pipe.to(DEVICE)
|
25 |
|
26 |
examples = [[Image.open("cat.jpg"), "a tiger", 0, 0.38, 0.5, 0.9, 28, 28, 789385745, False]]
|