Update app.py

app.py

@@ -35,7 +35,7 @@ import torch
 from diffusers import FluxPipeline, FluxTransformer2DModel, GGUFQuantizationConfig
 
 ckpt_path = (
-    "https://huggingface.co/city96/FLUX.1-dev-gguf/blob/main/flux1-schnell-Q3_K_S.gguf"
+    "https://huggingface.co/martintomov/Hyper-FLUX.1-dev-gguf/resolve/main/hyper-flux-16step-Q3_K_M.gguf"
 )
 transformer = FluxTransformer2DModel.from_single_file(
     ckpt_path,
@@ -43,12 +43,12 @@ transformer = FluxTransformer2DModel.from_single_file(
     torch_dtype=torch.bfloat16,
 )
 pipe = FluxPipeline.from_pretrained(
-    "black-forest-labs/FLUX.1-dev",
+    "black-forest-labs/FLUX.1-schnell",
     transformer=transformer,
     torch_dtype=torch.bfloat16,
 )
-
-pipe = FluxPipeline.from_pretrained("flux1-schnell-Q3_K_S.gguf")
+# https://huggingface.co/martintomov/Hyper-FLUX.1-dev-gguf/resolve/main/hyper-flux-16step-Q3_K_M.gguf
+#pipe = FluxPipeline.from_pretrained("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)