radames
/

sd-21-DPO-LoRA

Model card Files Files and versions Community

radames commited on Jan 9, 2024

Commit

c24038f

·

1 Parent(s): 92d8540

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -28,7 +28,7 @@ pipe = DiffusionPipeline.from_pretrained(
     # "stabilityai/stable-diffusion-2-1",
     torch_dtype=torch.float16, variant="fp16"
 )
-pipe.load_lora_weights("radames/dpo-lora-sd2.1")
 seed = 123123
 pipe.to("cuda")
 prompt = "portrait headshot professional of elon musk"

     # "stabilityai/stable-diffusion-2-1",
     torch_dtype=torch.float16, variant="fp16"
 )
+pipe.load_lora_weights("stable-diffusion-2-1-DPO-LoRA")
 seed = 123123
 pipe.to("cuda")
 prompt = "portrait headshot professional of elon musk"