justinpinkney
commited on
Commit
·
d1ff731
1
Parent(s):
8ee3c60
Update sd-superres-config.yaml
Browse files- sd-superres-config.yaml +1 -3
sd-superres-config.yaml
CHANGED
@@ -31,7 +31,6 @@ model:
|
|
31 |
params:
|
32 |
embed_dim: 4
|
33 |
monitor: val/rec_loss
|
34 |
-
ckpt_path: "models/first_stage_models/kl-f8/model.ckpt"
|
35 |
ddconfig:
|
36 |
double_z: true
|
37 |
z_channels: 4
|
@@ -82,7 +81,6 @@ model:
|
|
82 |
params:
|
83 |
embed_dim: 4
|
84 |
monitor: val/rec_loss
|
85 |
-
ckpt_path: "models/first_stage_models/kl-f8/model.ckpt"
|
86 |
ddconfig:
|
87 |
double_z: true
|
88 |
z_channels: 4
|
@@ -108,7 +106,7 @@ model:
|
|
108 |
data:
|
109 |
target: ldm.data.laion.WebDataModuleFromConfig
|
110 |
params:
|
111 |
-
tar_base:
|
112 |
batch_size: 3
|
113 |
num_workers: 2
|
114 |
multinode: True
|
|
|
31 |
params:
|
32 |
embed_dim: 4
|
33 |
monitor: val/rec_loss
|
|
|
34 |
ddconfig:
|
35 |
double_z: true
|
36 |
z_channels: 4
|
|
|
81 |
params:
|
82 |
embed_dim: 4
|
83 |
monitor: val/rec_loss
|
|
|
84 |
ddconfig:
|
85 |
double_z: true
|
86 |
z_channels: 4
|
|
|
106 |
data:
|
107 |
target: ldm.data.laion.WebDataModuleFromConfig
|
108 |
params:
|
109 |
+
tar_base: laion/improved_aesthetics_6plus/ims
|
110 |
batch_size: 3
|
111 |
num_workers: 2
|
112 |
multinode: True
|