neggles commited on
Commit
0f11131
1 Parent(s): bed8171

update vq-f8 config yamls to point at taming

Browse files
Files changed (2) hide show
  1. vq-f8-n256/config.yaml +1 -1
  2. vq-f8/config.yaml +1 -1
vq-f8-n256/config.yaml CHANGED
@@ -1,6 +1,6 @@
1
  model:
2
  base_learning_rate: 4.5e-06
3
- target: ldm.models.autoencoder.VQModel
4
  params:
5
  embed_dim: 4
6
  n_embed: 256
 
1
  model:
2
  base_learning_rate: 4.5e-06
3
+ target: taming.models.vqgan.VQModel
4
  params:
5
  embed_dim: 4
6
  n_embed: 256
vq-f8/config.yaml CHANGED
@@ -1,6 +1,6 @@
1
  model:
2
  base_learning_rate: 4.5e-06
3
- target: ldm.models.autoencoder.VQModel
4
  params:
5
  embed_dim: 4
6
  n_embed: 16384
 
1
  model:
2
  base_learning_rate: 4.5e-06
3
+ target: taming.models.vqgan.VQModel
4
  params:
5
  embed_dim: 4
6
  n_embed: 16384