Update app.py

app.py

@@ -741,7 +741,7 @@ def get_image(image) -> torch.Tensor | None:
 from huggingface_hub import hf_hub_download
 from safetensors.torch import load_file
 
-sd = load_file(hf_hub_download(repo_id="lllyasviel/flux1-dev-bnb-nf4", filename="flux1-dev-bnb-nf4.safetensors"))
+sd = load_file(hf_hub_download(repo_id="lllyasviel/flux1-dev-bnb-nf4", filename="flux1-dev-bnb-nf4-v2.safetensors"))
 sd = {k.replace("model.diffusion_model.", ""): v for k, v in sd.items() if "model.diffusion_model" in k}
 model = Flux().to(dtype=torch.bfloat16, device="cuda")
 result = model.load_state_dict(sd)