diff --git a/configs/INITIAL_MODELS.yaml b/configs/INITIAL_MODELS.yaml index ea43280340..1cdee3ca92 100644 --- a/configs/INITIAL_MODELS.yaml +++ b/configs/INITIAL_MODELS.yaml @@ -41,6 +41,8 @@ waifu-diffusion-1.4: description: Latest waifu diffusion 1.4 (diffusers version) format: diffusers repo_id: hakurei/waifu-diffusion + vae: + repo_id: stabilityai/sd-vae-ft-mse recommended: True waifu-diffusion-1.3: description: Stable Diffusion 1.4 fine tuned on anime-styled images (ckpt version) (4.27 GB) @@ -77,6 +79,8 @@ anything-4.0: description: High-quality, highly detailed anime style images with just a few prompts format: diffusers repo_id: andite/anything-v4.0 + vae: + repo_id: stabilityai/sd-vae-ft-mse recommended: False papercut-1.0: description: SD 1.5 fine-tuned for papercut art (use "PaperCut" in your prompts) (2.13 GB)