Spaces:

RED-AIGC
/

FLUX-TDD-BETA

Running on Zero

jychen0828 commited on Dec 29, 2024

Commit

e8d1004

verified ·

1 Parent(s): c4a76bb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -32,7 +32,7 @@ if not path.exists(cache_path):
 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_lora_weights.safetensors"),adapter_name="TDD")
-pipe.load_lora_weights(hf_hub_download("RED-AIGC/TDD", "FLUX.1-dev_tdd_adv_lora_weights1.safetensors"),adapter_name="TDD_adv")
 pipe.fuse_lora(lora_scale=0.125)
 pipe.to("cuda")

 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_lora_weights.safetensors"),adapter_name="TDD")
+pipe.load_lora_weights(hf_hub_download("RED-AIGC/TDD", "FLUX.1-dev_tdd_adv_lora_weights.safetensors"),adapter_name="TDD_adv")
 pipe.fuse_lora(lora_scale=0.125)
 pipe.to("cuda")