Hugging Face's logo

ezlee258258
/
Inversion-DPO

Text-to-Image
Diffusers
Safetensors
stable-diffusion-xl
stable-diffusion
inversion
dpo
fine-tuned
Model card Files
xet
 Community
1
ezlee258258 commited on
Commit
b771528
·
verified ·
1 Parent(s): 4acf8cb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -4
README.md CHANGED
@@ -31,15 +31,13 @@ import torch
31
  # Load the fine-tuned UNet
32
  unet = UNet2DConditionModel.from_pretrained(
33
  "ezlee258258/Inversion-DPO",
34
- subfolder="unet",
35
- torch_dtype=torch.float16
36
  )
37
 
38
  # Load the pipeline with the fine-tuned UNet
39
  pipe = StableDiffusionXLPipeline.from_pretrained(
40
  "stabilityai/stable-diffusion-xl-base-1.0",
41
- unet=unet,
42
- torch_dtype=torch.float16
43
  )
44
  pipe = pipe.to("cuda")
45
 
 
31
  # Load the fine-tuned UNet
32
  unet = UNet2DConditionModel.from_pretrained(
33
  "ezlee258258/Inversion-DPO",
34
+ subfolder="unet"
 
35
  )
36
 
37
  # Load the pipeline with the fine-tuned UNet
38
  pipe = StableDiffusionXLPipeline.from_pretrained(
39
  "stabilityai/stable-diffusion-xl-base-1.0",
40
+ unet=unet
 
41
  )
42
  pipe = pipe.to("cuda")
43