Hugging Face's logo

jacklishufan
/
diffusion-kto

Diffusers
Safetensors
Model card Files Community
1

Update README.md

#1
by akashgokul - opened
base: refs/heads/main
←
from: refs/pr/1
Discussion Files changed
+2
-2
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -9,11 +9,11 @@ library_name: diffusers
9
  </p>
10
 
11
 
12
- This model is fine-tuned from stable-diffusion-xl-base-1.0 on offline human preference data pickapic_v2 using KTO.
13
 
14
 
15
  ### Usage
16
- ```
17
  import torch
18
  from diffusers import AutoencoderKL, UNet2DConditionModel, DiffusionPipeline
19
  vae_path = model_name = "runwayml/stable-diffusion-v1-5"
 
9
  </p>
10
 
11
 
12
+ This model is fine-tuned from Stable Diffusion v1-5 on Pick-a-Pic v2 dataset using KTO.
13
 
14
 
15
  ### Usage
16
+ ```python
17
  import torch
18
  from diffusers import AutoencoderKL, UNet2DConditionModel, DiffusionPipeline
19
  vae_path = model_name = "runwayml/stable-diffusion-v1-5"