Spaces:
Runtime error
Runtime error
Commit
·
4b0c1a8
1
Parent(s):
c260e72
Update 2e5_debug_gpt_firstframe_posmap.yaml
Browse files
2e5_debug_gpt_firstframe_posmap.yaml
CHANGED
@@ -2,7 +2,7 @@ save_path: saved_fixcursor_lr2e5_debug_gpt_firstframe_posmap
|
|
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
|