multimodalart HF staff commited on
Commit
881ece2
1 Parent(s): ea751d9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -18,8 +18,8 @@ from diffusers.models import AutoencoderKL
18
  from diffusers import StableDiffusionPipeline
19
 
20
  model = "CompVis/stable-diffusion-v1-4"
21
- vae = AutoencoderKL("stabilityai/sd-vae-ft-mse")
22
- pipe = StableDiffusionPipeline(model, vae=vae)
23
  ```
24
 
25
  ## Decoder Finetuning
 
18
  from diffusers import StableDiffusionPipeline
19
 
20
  model = "CompVis/stable-diffusion-v1-4"
21
+ vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse")
22
+ pipe = StableDiffusionPipeline.from_pretrained(model, vae=vae)
23
  ```
24
 
25
  ## Decoder Finetuning