Update app.py
Browse files
app.py
CHANGED
|
@@ -31,7 +31,6 @@ JS = """function () {
|
|
| 31 |
|
| 32 |
if torch.cuda.is_available():
|
| 33 |
pipe = FluxPipeline.from_pretrained(model, torch_dtype=torch.bfloat16)
|
| 34 |
-
pipe.load_lora_weights(hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-8steps-lora.safetensors"))
|
| 35 |
pipe.load_lora_weights("Shakker-Labs/FLUX.1-dev-LoRA-blended-realistic-illustration", weight_name="FLUX-dev-lora-blended_realistic_illustration.safetensors")
|
| 36 |
pipe.fuse_lora(lora_scale=1.0)
|
| 37 |
pipe.to(device="cuda", dtype=torch.bfloat16)
|
|
|
|
| 31 |
|
| 32 |
if torch.cuda.is_available():
|
| 33 |
pipe = FluxPipeline.from_pretrained(model, torch_dtype=torch.bfloat16)
|
|
|
|
| 34 |
pipe.load_lora_weights("Shakker-Labs/FLUX.1-dev-LoRA-blended-realistic-illustration", weight_name="FLUX-dev-lora-blended_realistic_illustration.safetensors")
|
| 35 |
pipe.fuse_lora(lora_scale=1.0)
|
| 36 |
pipe.to(device="cuda", dtype=torch.bfloat16)
|