Merge pull request #12 from JayChen7777/main

Update README.md
RedAIGC · Dec 26, 2024 · 686a70d · 686a70d
2 parents cb8776c + a3be527
commit 686a70d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.md b/README.md
@@ -62,7 +62,7 @@ from huggingface_hub import hf_hub_download
 from diffusers import FluxPipeline
 
 pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
-pipe.load_lora_weights(hf_hub_download("RED-AIGC/TDD", "TDD-FLUX.1-dev-lora-beta.safetensors"))
+pipe.load_lora_weights(hf_hub_download("RED-AIGC/TDD", "FLUX.1-dev_tdd_lora_weights.safetensors"))
 pipe.fuse_lora(lora_scale=0.125)
 pipe.to("cuda")