ysharma HF staff commited on
Commit
bfb07fc
1 Parent(s): bca52a3
Files changed (1) hide show
  1. txt2img-1p4B-eval.yaml +4 -4
txt2img-1p4B-eval.yaml CHANGED
@@ -1,6 +1,6 @@
1
  model:
2
  base_learning_rate: 5.0e-05
3
- target: ldm.models.diffusion.ddpm.LatentDiffusion
4
  params:
5
  linear_start: 0.00085
6
  linear_end: 0.012
@@ -18,7 +18,7 @@ model:
18
  use_ema: False
19
 
20
  unet_config:
21
- target: ldm.modules.diffusionmodules.openaimodel.UNetModel
22
  params:
23
  image_size: 32
24
  in_channels: 4
@@ -42,7 +42,7 @@ model:
42
  legacy: False
43
 
44
  first_stage_config:
45
- target: ldm.models.autoencoder.AutoencoderKL
46
  params:
47
  embed_dim: 4
48
  monitor: val/rec_loss
@@ -65,7 +65,7 @@ model:
65
  target: torch.nn.Identity
66
 
67
  cond_stage_config:
68
- target: ldm.modules.encoders.modules.BERTEmbedder
69
  params:
70
  n_embed: 1280
71
  n_layer: 32
 
1
  model:
2
  base_learning_rate: 5.0e-05
3
+ target: ddpm.LatentDiffusion
4
  params:
5
  linear_start: 0.00085
6
  linear_end: 0.012
 
18
  use_ema: False
19
 
20
  unet_config:
21
+ target: openaimodel.UNetModel
22
  params:
23
  image_size: 32
24
  in_channels: 4
 
42
  legacy: False
43
 
44
  first_stage_config:
45
+ target: autoencoder.AutoencoderKL
46
  params:
47
  embed_dim: 4
48
  monitor: val/rec_loss
 
65
  target: torch.nn.Identity
66
 
67
  cond_stage_config:
68
+ target: modules.BERTEmbedder
69
  params:
70
  n_embed: 1280
71
  n_layer: 32