Hyper-FLUX-8Steps-LoRA_rank1

Running on Zero

bdsqlsz commited on Sep 11, 2024

Commit

18e29bb

verified ·

1 Parent(s): f1baa6b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -30,7 +30,7 @@ class timer:
 if not path.exists(cache_path):
     os.makedirs(cache_path, exist_ok=True)
-pipe = FluxPipeline.from_single_file("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)

 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)