From db580ccefd55515e90e29ad44e19a62ccf40c8a6 Mon Sep 17 00:00:00 2001 From: David Wager Date: Thu, 1 Sep 2022 19:02:57 +0100 Subject: [PATCH] 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. --- configs/models.yaml | 10 ++++++++++ 1 file changed, 10 insertions(+) create mode 100644 configs/models.yaml diff --git a/configs/models.yaml b/configs/models.yaml new file mode 100644 index 0000000000..cb4f6b2770 --- /dev/null +++ b/configs/models.yaml @@ -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 \ No newline at end of file