K00B404 commited on
Commit
d2e8c36
·
verified ·
1 Parent(s): 956637f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -29,8 +29,10 @@ class timer:
29
 
30
  if not path.exists(cache_path):
31
  os.makedirs(cache_path, exist_ok=True)
 
32
 
33
- pipe = FluxPipeline.from_pretrained("https://huggingface.co/city96/FLUX.1-schnell-gguf/resolve/main/flux1-schnell-Q3_K_S.gguf")
 
34
  pipe.load_lora_weights(hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-8steps-lora.safetensors"))
35
  pipe.fuse_lora(lora_scale=0.125)
36
  pipe.to(device="cuda", dtype=torch.bfloat16)
 
29
 
30
  if not path.exists(cache_path):
31
  os.makedirs(cache_path, exist_ok=True)
32
+ from huggingface_hub import hf_hub_download
33
 
34
+ hf_hub_download(repo_id="city96/FLUX.1-schnell-gguf", filename="flux1-schnell-Q3_K_S.gguf")
35
+ pipe = FluxPipeline.from_pretrained("flux1-schnell-Q3_K_S.gguf")
36
  pipe.load_lora_weights(hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-8steps-lora.safetensors"))
37
  pipe.fuse_lora(lora_scale=0.125)
38
  pipe.to(device="cuda", dtype=torch.bfloat16)