fh2c1
/

Anonymize-Anyone

StableDiffusionPipeline

Model card Files Files and versions Metrics Training metrics Community

fh2c1 commited on Feb 11

Commit

8baeb66

·

verified ·

1 Parent(s): 12feda3

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -41,7 +41,7 @@ sd_pipe = StableDiffusionInpaintPipeline.from_pretrained(
     torch_dtype=torch.float16,
     safety_checker=None,
 ).to(device)
-sd_pipe.load_lora_weights("fh2c1/SimPO-LoRA", adapter_name="SimPO")
 sd_pipe.set_adapters(["SimPO"], adapter_weights=[0.5])
 def generate_image(image_path, mask_path, prompt, negative_prompt, pipe, seed):

     torch_dtype=torch.float16,
     safety_checker=None,
 ).to(device)
+sd_pipe.load_lora_weights("fh2c1/SimPO-LoRA-1.2", adapter_name="SimPO")
 sd_pipe.set_adapters(["SimPO"], adapter_weights=[0.5])
 def generate_image(image_path, mask_path, prompt, negative_prompt, pipe, seed):