radames
/

sd-21-DPO-LoRA

Model card Files Files and versions Community

radames commited on Jan 12

Commit

c2bd4be

•

1 Parent(s): 70eb729

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -25,7 +25,7 @@ pipe = DiffusionPipeline.from_pretrained(
     torch_dtype=torch.float16, variant="fp16"
 )
 pipe.to("cuda")
-pipe.load_lora_weights("radames/stable-diffusion-2-1-DPO-LoRA", adapter_name="dpo-lora-sd21")
 pipe.set_adapters(["dpo-lora-sd21"], adapter_weights=[1.0]) # you can play with adapter_weights to increase the effect of the LoRA model
 seed = 123123
 prompt = "portrait headshot professional of elon musk"

     torch_dtype=torch.float16, variant="fp16"
 )
 pipe.to("cuda")
+pipe.load_lora_weights("radames/sd-21-DPO-LoRA", adapter_name="dpo-lora-sd21")
 pipe.set_adapters(["dpo-lora-sd21"], adapter_weights=[1.0]) # you can play with adapter_weights to increase the effect of the LoRA model
 seed = 123123
 prompt = "portrait headshot professional of elon musk"