ckpt

2022-06-17 23:05:03 +00:00 · 2022-06-17 23:05:03 +00:00 · 12d2aaf50c
parent c790c34e21
commit 12d2aaf50c
1 changed files with 1 additions and 2 deletions
--- 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