From 12d2aaf50c320f1bc500d51c339103efd87aa4f7 Mon Sep 17 00:00:00 2001 From: Patrick Esser Date: Fri, 17 Jun 2022 23:05:03 +0000 Subject: [PATCH] ckpt --- .../txt2img-multinode-clip-encoder-f16-768.yaml | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/configs/stable-diffusion/txt2img-multinode-clip-encoder-f16-768.yaml b/configs/stable-diffusion/txt2img-multinode-clip-encoder-f16-768.yaml index 184905a..707eda9 100644 --- a/configs/stable-diffusion/txt2img-multinode-clip-encoder-f16-768.yaml +++ b/configs/stable-diffusion/txt2img-multinode-clip-encoder-f16-768.yaml @@ -16,7 +16,7 @@ model: monitor: val/loss_simple_ema scale_factor: 0.22765929 # magic number - ckpt_path: # TODO: add + ckpt_path: "/home/mchorse/stable-diffusion-ckpts/256f16-2022-06-15-216k-pruned.ckpt" scheduler_config: # 10000 warmup steps target: ldm.lr_scheduler.LambdaLinearScheduler @@ -49,7 +49,6 @@ model: params: embed_dim: 16 monitor: val/rec_loss - ckpt_path: "models/first_stage_models/kl-f16/model.ckpt" ddconfig: double_z: True z_channels: 16