models.yaml 835 B

1234567891011121314151617181920212223
  1. # This file describes the alternative machine learning models
  2. # available to InvokeAI script.
  3. #
  4. # To add a new model, follow the examples below. Each
  5. # model requires a model config file, a weights file,
  6. # and the width and height of the images it
  7. # was trained on.
  8. stable-diffusion-1.5:
  9. description: Stable Diffusion version 1.5
  10. weights: /data/StableDiffusion/v1-5-pruned-emaonly.ckpt
  11. vae: /data/VAE/vae-ft-mse-840000-ema-pruned.ckpt
  12. config: ./configs/stable-diffusion/v1-inference.yaml
  13. width: 512
  14. height: 512
  15. default: true
  16. inpainting-1.5:
  17. description: RunwayML SD 1.5 model optimized for inpainting
  18. weights: /data/StableDiffusion/sd-v1-5-inpainting.ckpt
  19. vae: /data/VAE/vae-ft-mse-840000-ema-pruned.ckpt
  20. config: ./configs/stable-diffusion/v1-inpainting-inference.yaml
  21. width: 512
  22. height: 512
  23. default: false