benjamin-paine
/

sd-dpo-offsets

Model card Files Files and versions Community

benjamin-paine commited on Dec 24, 2023

Commit

1669423

•

1 Parent(s): 3a118d5

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -104,7 +104,7 @@ image = pipe(prompt, guidance_scale=7.5, generator=generator).images[0]
 image.save("cats_playing_chess_xl.png")
 # download DPO offsets
-dpo_offset_path = huggingface_hub.hf_hub_download("benjamin-paine/sd-dpo-offsets", "sd_v15_unet_dpo_offset.safetensors")
 # inject
 inject_dpo(pipe.unet, dpo_offset_path, device)

 image.save("cats_playing_chess_xl.png")
 # download DPO offsets
+dpo_offset_path = huggingface_hub.hf_hub_download("benjamin-paine/sd-dpo-offsets", "sd_xl_unet_dpo_offset.safetensors")
 # inject
 inject_dpo(pipe.unet, dpo_offset_path, device)