Emaad commited on
Commit
36ee724
1 Parent(s): 4d6deaa

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