mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
531f596bd1
- applied inpainting parameters recommended by @kyle0654 - results are aesthetically pleasing - Closes #1319
44 lines
1.8 KiB
YAML
44 lines
1.8 KiB
YAML
# This file describes the alternative machine learning models
|
|
# available to InvokeAI 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/sd-v1-4.ckpt
|
|
vae: ./models/ldm/stable-diffusion-v1/vae-ft-mse-840000-ema-pruned.ckpt
|
|
description: The original Stable Diffusion version 1.4 weight file (4.27 GB)
|
|
width: 512
|
|
height: 512
|
|
stable-diffusion-1.5:
|
|
description: The newest Stable Diffusion version 1.5 weight file (4.27 GB)
|
|
weights: ./models/ldm/stable-diffusion-v1/v1-5-pruned-emaonly.ckpt
|
|
config: ./configs/stable-diffusion/v1-inference.yaml
|
|
width: 512
|
|
height: 512
|
|
vae: ./models/ldm/stable-diffusion-v1/vae-ft-mse-840000-ema-pruned.ckpt
|
|
default: true
|
|
inpainting-1.5:
|
|
description: RunwayML SD 1.5 model optimized for inpainting (4.27 GB)
|
|
weights: ./models/ldm/stable-diffusion-v1/sd-v1-5-inpainting.ckpt
|
|
config: ./configs/stable-diffusion/v1-inpainting-inference.yaml
|
|
width: 512
|
|
height: 512
|
|
vae: ./models/ldm/stable-diffusion-v1/vae-ft-mse-840000-ema-pruned.ckpt
|
|
waifu-diffusion-1.3:
|
|
description: Stable Diffusion 1.4 fine tuned on anime-styled images (4.27)
|
|
weights: ./models/ldm/stable-diffusion-v1/model-epoch09-float32.ckpt
|
|
config: ./configs/stable-diffusion/v1-inference.yaml
|
|
width: 512
|
|
height: 512
|
|
vae: ./models/ldm/stable-diffusion-v1/vae-ft-mse-840000-ema-pruned.ckpt
|
|
smirking-face-1.0:
|
|
description: Stable Diffusion 1.4 fine tuned on human figures (4.27)
|
|
weights: ./models/ldm/stable-diffusion-v1/SF_EB_1.0_ema_vae.ckpt
|
|
config: ./configs/stable-diffusion/v1-inference.yaml
|
|
width: 512
|
|
height: 512
|
|
vae: ./models/ldm/stable-diffusion-v1/vae-ft-mse-840000-ema-pruned.ckpt
|