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
|