update vq-f8 config yamls to point at taming
Browse files- vq-f8-n256/config.yaml +1 -1
- 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:
|
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:
|
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
|