RED-AIGC
/

TDD

jychen0828 commited on Jan 6, 2025

Commit

285176b

verified ·

1 Parent(s): 740a2ad

Update README.md

Files changed (1) hide show

README.md CHANGED Viewed

@@ -30,7 +30,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.fuse_lora(lora_scale=0.125)
 pipe.to("cuda")

 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", "FLUX.1-dev_tdd_adv_lora_weights.safetensors"))
 pipe.fuse_lora(lora_scale=0.125)
 pipe.to("cuda")