Delete models
Browse files
models/CELL-E_2_HPA_480.ckpt
DELETED
@@ -1,3 +0,0 @@
|
|
1 |
-
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:14b445eedda4f46e67c7d848df3301363e9034d752591a1275b47eaf483e0cac
|
3 |
-
size 4731482483
|
|
|
|
|
|
|
|
models/CELL-E_2_HPA_480.yaml
DELETED
@@ -1,25 +0,0 @@
|
|
1 |
-
model:
|
2 |
-
learning_rate: 0.0003
|
3 |
-
target: celle_main.CELLE_trainer
|
4 |
-
params:
|
5 |
-
ckpt_path: CELL-E_2_HPA_480.ckpt
|
6 |
-
condition_model_path:
|
7 |
-
condition_config_path: nucleus_vqgan.yaml
|
8 |
-
vqgan_model_path:
|
9 |
-
vqgan_config_path: threshold_vqgan.yaml
|
10 |
-
image_key: threshold
|
11 |
-
num_images: 2
|
12 |
-
dim: 480
|
13 |
-
num_text_tokens: 33
|
14 |
-
text_seq_len: 1000
|
15 |
-
depth: 68
|
16 |
-
heads: 16
|
17 |
-
dim_head: 64
|
18 |
-
attn_dropout: 0.1
|
19 |
-
ff_dropout: 0.1
|
20 |
-
attn_types: full
|
21 |
-
rotary_emb: true
|
22 |
-
fixed_embedding: true
|
23 |
-
text_embedding: esm2
|
24 |
-
loss_img_weight: 1
|
25 |
-
loss_cond_weight: 1
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
models/CELL-E_2_HPA_Finetuned_480.ckpt
DELETED
@@ -1,3 +0,0 @@
|
|
1 |
-
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:152dfd30c8adfe83868ac0ffea4d075d25d11f24369f34482f79a3db4524d1be
|
3 |
-
size 4731482675
|
|
|
|
|
|
|
|
models/CELL-E_2_HPA_Finetuned_480.yaml
DELETED
@@ -1,25 +0,0 @@
|
|
1 |
-
model:
|
2 |
-
learning_rate: 0.0003
|
3 |
-
target: celle_main.CELLE_trainer
|
4 |
-
params:
|
5 |
-
ckpt_path: CELL-E_2_HPA_Finetuned_480.ckpt
|
6 |
-
condition_model_path:
|
7 |
-
condition_config_path: nucleus_vqgan.yaml
|
8 |
-
vqgan_model_path:
|
9 |
-
vqgan_config_path: threshold_vqgan.yaml
|
10 |
-
image_key: threshold
|
11 |
-
num_images: 2
|
12 |
-
dim: 480
|
13 |
-
num_text_tokens: 33
|
14 |
-
text_seq_len: 1000
|
15 |
-
depth: 68
|
16 |
-
heads: 16
|
17 |
-
dim_head: 64
|
18 |
-
attn_dropout: 0.1
|
19 |
-
ff_dropout: 0.1
|
20 |
-
attn_types: full
|
21 |
-
rotary_emb: true
|
22 |
-
fixed_embedding: true
|
23 |
-
text_embedding: esm2
|
24 |
-
loss_img_weight: 1
|
25 |
-
loss_cond_weight: 1
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
models/nucleus_vqgan.yaml
DELETED
@@ -1,33 +0,0 @@
|
|
1 |
-
model:
|
2 |
-
base_learning_rate: 4.5e-06
|
3 |
-
target: taming.models.vqgan.VQModel
|
4 |
-
params:
|
5 |
-
image_key: nucleus
|
6 |
-
ckpt_path:
|
7 |
-
embed_dim: 256
|
8 |
-
n_embed: 512
|
9 |
-
ddconfig:
|
10 |
-
double_z: false
|
11 |
-
z_channels: 256
|
12 |
-
resolution: 256
|
13 |
-
in_channels: 1
|
14 |
-
out_ch: 1
|
15 |
-
ch: 128
|
16 |
-
ch_mult:
|
17 |
-
- 1
|
18 |
-
- 1
|
19 |
-
- 2
|
20 |
-
- 2
|
21 |
-
- 4
|
22 |
-
num_res_blocks: 2
|
23 |
-
attn_resolutions:
|
24 |
-
- 16
|
25 |
-
dropout: 0.0
|
26 |
-
lossconfig:
|
27 |
-
target: taming.modules.losses.vqperceptual.VQLPIPSWithDiscriminator
|
28 |
-
params:
|
29 |
-
disc_conditional: false
|
30 |
-
disc_in_channels: 1
|
31 |
-
disc_start: 50000
|
32 |
-
disc_weight: 0.2
|
33 |
-
codebook_weight: 1.0
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
models/threshold_vqgan.yaml
DELETED
@@ -1,33 +0,0 @@
|
|
1 |
-
model:
|
2 |
-
base_learning_rate: 4.5e-05
|
3 |
-
target: taming.models.vqgan.VQModel
|
4 |
-
params:
|
5 |
-
image_key: threshold
|
6 |
-
ckpt_path:
|
7 |
-
embed_dim: 256
|
8 |
-
n_embed: 512
|
9 |
-
ddconfig:
|
10 |
-
double_z: false
|
11 |
-
z_channels: 256
|
12 |
-
resolution: 256
|
13 |
-
in_channels: 1
|
14 |
-
out_ch: 1
|
15 |
-
ch: 128
|
16 |
-
ch_mult:
|
17 |
-
- 1
|
18 |
-
- 1
|
19 |
-
- 2
|
20 |
-
- 2
|
21 |
-
- 4
|
22 |
-
num_res_blocks: 2
|
23 |
-
attn_resolutions:
|
24 |
-
- 16
|
25 |
-
dropout: 0.0
|
26 |
-
lossconfig:
|
27 |
-
target: taming.modules.losses.vqperceptual.VQLPIPSWithDiscriminator
|
28 |
-
params:
|
29 |
-
disc_conditional: false
|
30 |
-
disc_in_channels: 1
|
31 |
-
disc_start: 100000
|
32 |
-
disc_weight: 0.2
|
33 |
-
codebook_weight: 1.0
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|