InvokeAI/configs/models.yaml

# This file describes the alternative machine learning models
# available to the dream script.
#
# To add a new model, follow the examples below. Each
# model requires a model config file, a weights file,
# and the width and height of the images it
# was trained on.
stable-diffusion-1.4:
  config: configs/stable-diffusion/v1-inference.yaml
  weights: models/ldm/stable-diffusion-v1/model.ckpt
  vae: models/ldm/stable-diffusion-v1/vae-ft-mse-840000-ema-pruned.ckpt
  description: Stable Diffusion inference model version 1.4
  width: 512
  height: 512
stable-diffusion-1.5:
  config: configs/stable-diffusion/v1-inference.yaml
  weights: models/ldm/stable-diffusion-v1/v1-5-pruned-emaonly.ckpt
  description: Stable Diffusion inference model version 1.5
  width: 512
  height: 512
  vae: models/ldm/stable-diffusion-v1/vae-ft-mse-840000-ema-pruned.ckpt
  default: true
add ability to specify location of config file (models.yaml) 2022-09-03 15:36:04 +00:00			`# This file describes the alternative machine learning models`
minor fixes to inpaint code 1. If tensors are passed to inpaint as init_image and/or init_mask, then the post-generation image fixup code will be skipped. 2. Post-generation image fixup will work with either a black and white "L" or "RGB" mask, or an "RGBA" mask. 2022-10-23 02:28:54 +00:00			`# available to the dream script.`
add ability to specify location of config file (models.yaml) 2022-09-03 15:36:04 +00:00			`#`
			`# To add a new model, follow the examples below. Each`
			`# model requires a model config file, a weights file,`
			`# and the width and height of the images it`
			`# was trained on.`
Create models.yaml models.yaml can serve as a base for expanding our support for other versions of Latent/Stable Diffusion. Contained are parameters for default width/height, as well as where to find the config and weights for this model. Adding a new model is as simple as adding to this file. 2022-09-01 18:02:57 +00:00			`stable-diffusion-1.4:`
minor fixes to inpaint code 1. If tensors are passed to inpaint as init_image and/or init_mask, then the post-generation image fixup code will be skipped. 2. Post-generation image fixup will work with either a black and white "L" or "RGB" mask, or an "RGBA" mask. 2022-10-23 02:28:54 +00:00			`config: configs/stable-diffusion/v1-inference.yaml`
			`weights: models/ldm/stable-diffusion-v1/model.ckpt`
			`vae: models/ldm/stable-diffusion-v1/vae-ft-mse-840000-ema-pruned.ckpt`
			`description: Stable Diffusion inference model version 1.4`
			`width: 512`
			`height: 512`
further improvements to model loading - code for committing config changes to models.yaml now in module rather than in invoke script - model marked "default" is now loaded if model not specified on command line - uncache changed models when edited, so that they reload properly - removed liaon from models.yaml and added stable-diffusion-1.5 2022-10-21 04:28:54 +00:00			`stable-diffusion-1.5:`
minor fixes to inpaint code 1. If tensors are passed to inpaint as init_image and/or init_mask, then the post-generation image fixup code will be skipped. 2. Post-generation image fixup will work with either a black and white "L" or "RGB" mask, or an "RGBA" mask. 2022-10-23 02:28:54 +00:00			`config: configs/stable-diffusion/v1-inference.yaml`
			`weights: models/ldm/stable-diffusion-v1/v1-5-pruned-emaonly.ckpt`
			`description: Stable Diffusion inference model version 1.5`
			`width: 512`
			`height: 512`
			`vae: models/ldm/stable-diffusion-v1/vae-ft-mse-840000-ema-pruned.ckpt`
			`default: true`