Update README.md

README.md

@@ -206,7 +206,7 @@ from diffusers import (
 
 # Load VAE component
 vae = AutoencoderKL.from_pretrained(
-    "cagliostrolab/animagine-xl-3.0", 
+    "madebyollin/sdxl-vae-fp16-fix", 
     torch_dtype=torch.float16
 )
 
@@ -348,6 +348,8 @@ This model supports generating images at the following dimensions:
 | **Image Resolution**  | 1024                    | 1024                    |
 | **Bucket Resolution** | 2048 x 512              | 2048 x 512              |
 
+Source code and training config are available here: https://github.com/cagliostrolab/sd-scripts/tree/main/notebook 
+
 ## Limitations
 
 While "Animagine XL 3.0" represents a significant advancement in anime text-to-image generation, it's important to acknowledge its limitations to understand its best use cases and potential areas for future improvement.