Hyper-FLUX-8Steps-LoRA_CPU

Runtime error

K00B404 commited on Jan 30

Commit

956637f

verified ·

1 Parent(s): 6a35d52

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_pretrained("city96/FLUX.1-schnell-gguf", torch_dtype=torch.bfloat16)
 pipe.load_lora_weights(hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-8steps-lora.safetensors"))
 pipe.fuse_lora(lora_scale=0.125)
 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/city96/FLUX.1-schnell-gguf/resolve/main/flux1-schnell-Q3_K_S.gguf")
 pipe.load_lora_weights(hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-8steps-lora.safetensors"))
 pipe.fuse_lora(lora_scale=0.125)
 pipe.to(device="cuda", dtype=torch.bfloat16)