diff --git a/configs/models.yaml b/configs/models.yaml
index 3daa80246f..77963e82f5 100644
--- a/configs/models.yaml
+++ b/configs/models.yaml
@@ -6,9 +6,9 @@
 # and the width and height of the images it
 # was trained on.
 stable-diffusion-1.4:
-  config:  configs/stable-diffusion/v1-inference.yaml
+  config: configs/stable-diffusion/v1-inference.yaml
   weights: models/ldm/stable-diffusion-v1/model.ckpt
-  description: Stable Diffusion inference model version 1.4
+  vae: models/ldm/stable-diffusion-v1/vae-ft-mse-840000-ema-pruned.ckpt
   width: 512
   height: 512
   default: true