ckpt
This commit is contained in:
parent
c790c34e21
commit
12d2aaf50c
1 changed files with 1 additions and 2 deletions
|
@ -16,7 +16,7 @@ model:
|
||||||
monitor: val/loss_simple_ema
|
monitor: val/loss_simple_ema
|
||||||
scale_factor: 0.22765929 # magic number
|
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
|
scheduler_config: # 10000 warmup steps
|
||||||
target: ldm.lr_scheduler.LambdaLinearScheduler
|
target: ldm.lr_scheduler.LambdaLinearScheduler
|
||||||
|
@ -49,7 +49,6 @@ model:
|
||||||
params:
|
params:
|
||||||
embed_dim: 16
|
embed_dim: 16
|
||||||
monitor: val/rec_loss
|
monitor: val/rec_loss
|
||||||
ckpt_path: "models/first_stage_models/kl-f16/model.ckpt"
|
|
||||||
ddconfig:
|
ddconfig:
|
||||||
double_z: True
|
double_z: True
|
||||||
z_channels: 16
|
z_channels: 16
|
||||||
|
|
Loading…
Reference in a new issue