Hyper-FLUX-8Steps-LoRA_rank1

Sleeping

bdsqlsz commited on Sep 11, 2024

Commit

e822a67

verified ·

1 Parent(s): 18e29bb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -30,9 +30,9 @@ class timer:
 if not path.exists(cache_path):
     os.makedirs(cache_path, exist_ok=True)
-pipe = FluxPipeline.from_pretrained("https://huggingface.co/bdsqlsz/Hyper-Flux1-dev/blob/main/Hyper-flux1-dev.safetensors", torch_dtype=torch.bfloat16)
-#pipe.load_lora_weights(hf_hub_download("bdsqlsz/Hyper-Flux1-dev", "Hyper-FLUX.1-dev-8steps-lora_rank1.safetensors"))
-#pipe.fuse_lora(lora_scale=0.6)
 pipe.to(device="cuda", dtype=torch.bfloat16)
 with gr.Blocks(theme=gr.themes.Soft()) as demo:

 if not path.exists(cache_path):
     os.makedirs(cache_path, exist_ok=True)
+pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
+pipe.load_lora_weights(hf_hub_download("bdsqlsz/Hyper-Flux1-dev", "Hyper-FLUX.1-dev-8steps-lora_rank1.safetensors"))
+pipe.fuse_lora(lora_scale=0.3)
 pipe.to(device="cuda", dtype=torch.bfloat16)
 with gr.Blocks(theme=gr.themes.Soft()) as demo: