Spaces:
Runtime error
Runtime error
Commit
·
86bfabd
1
Parent(s):
3dc3e19
Update 2e5_debug_gpt_firstframe_posmap_longtrainh200.yaml
Browse files
2e5_debug_gpt_firstframe_posmap_longtrainh200.yaml
CHANGED
|
@@ -2,7 +2,7 @@ save_path: saved_fixcursor_lr2e5_debug_gpt_firstframe_posmap_longtrainh200
|
|
| 2 |
|
| 3 |
model:
|
| 4 |
base_learning_rate: 2.0e-05
|
| 5 |
-
target:
|
| 6 |
params:
|
| 7 |
linear_start: 0.0015
|
| 8 |
linear_end: 0.0195
|
|
@@ -20,7 +20,7 @@ model:
|
|
| 20 |
monitor: val/loss_simple_ema
|
| 21 |
|
| 22 |
unet_config:
|
| 23 |
-
target:
|
| 24 |
params:
|
| 25 |
image_size: 64
|
| 26 |
in_channels: 25
|
|
@@ -41,7 +41,7 @@ model:
|
|
| 41 |
transformer_depth: 1
|
| 42 |
context_dim: 768
|
| 43 |
first_stage_config:
|
| 44 |
-
target:
|
| 45 |
params:
|
| 46 |
embed_dim: 3
|
| 47 |
n_embed: 8192
|
|
@@ -65,7 +65,7 @@ model:
|
|
| 65 |
target: torch.nn.Identity
|
| 66 |
|
| 67 |
cond_stage_config:
|
| 68 |
-
target:
|
| 69 |
params:
|
| 70 |
n_embed: 768
|
| 71 |
n_layer: 12
|
|
|
|
| 2 |
|
| 3 |
model:
|
| 4 |
base_learning_rate: 2.0e-05
|
| 5 |
+
target: ldm.models.diffusion.ddpm.LatentDiffusion
|
| 6 |
params:
|
| 7 |
linear_start: 0.0015
|
| 8 |
linear_end: 0.0195
|
|
|
|
| 20 |
monitor: val/loss_simple_ema
|
| 21 |
|
| 22 |
unet_config:
|
| 23 |
+
target: ldm.modules.diffusionmodules.openaimodel.UNetModel
|
| 24 |
params:
|
| 25 |
image_size: 64
|
| 26 |
in_channels: 25
|
|
|
|
| 41 |
transformer_depth: 1
|
| 42 |
context_dim: 768
|
| 43 |
first_stage_config:
|
| 44 |
+
target: ldm.models.autoencoder.VQModelInterface
|
| 45 |
params:
|
| 46 |
embed_dim: 3
|
| 47 |
n_embed: 8192
|
|
|
|
| 65 |
target: torch.nn.Identity
|
| 66 |
|
| 67 |
cond_stage_config:
|
| 68 |
+
target: ldm.modules.encoders.modules.GPTEmbedder
|
| 69 |
params:
|
| 70 |
n_embed: 768
|
| 71 |
n_layer: 12
|