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.
This commit is contained in:
David Wager 2022-09-01 19:02:57 +01:00 committed by GitHub
parent 9e99fcbc16
commit db580ccefd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

10
configs/models.yaml Normal file
View File

@ -0,0 +1,10 @@
laion400m:
config: configs/latent-diffusion/txt2img-1p4B-eval.yaml
weights: models/ldm/text2img-large/model.ckpt
width: 256
height: 256
stable-diffusion-1.4:
config: configs/stable-diffusion/v1-inference.yaml
weights: models/ldm/stable-diffusion-v1.4/model.ckpt
width: 512
height: 512