Hugging Face's logo

suyc21
/
CellFlux

Diffusers
Model card Files
xet
Community
suyccc commited on
Commit
a8fab2e
·
1 Parent(s): ae20e34

upload ckpts

Browse files
Files changed (24) hide show
  1. checkpoints/cellflux/bbbc021/checkpoint.pth +3 -0
  2. checkpoints/cellflux/cpg0000/checkpoint.pth +3 -0
  3. checkpoints/cellflux/rxrx1/checkpoint.pth +3 -0
  4. checkpoints/impa/bbbc021/embeddings.ckpt +3 -0
  5. checkpoints/impa/bbbc021/nets.ckpt +3 -0
  6. checkpoints/impa/bbbc021/optims.ckpt +3 -0
  7. checkpoints/impa/cpg0000/embeddings.ckpt +3 -0
  8. checkpoints/impa/cpg0000/nets.ckpt +3 -0
  9. checkpoints/impa/cpg0000/optims.ckpt +3 -0
  10. checkpoints/impa/rxrx1/embeddings.ckpt +3 -0
  11. checkpoints/impa/rxrx1/nets.ckpt +3 -0
  12. checkpoints/impa/rxrx1/optims.ckpt +3 -0
  13. checkpoints/phendiff/bbbc021/model_index.json +12 -0
  14. checkpoints/phendiff/bbbc021/scheduler/scheduler_config.json +19 -0
  15. checkpoints/phendiff/bbbc021/unet/config.json +37 -0
  16. checkpoints/phendiff/bbbc021/unet/diffusion_pytorch_model.bin +3 -0
  17. checkpoints/phendiff/cpg0000/model_index.json +12 -0
  18. checkpoints/phendiff/cpg0000/scheduler/scheduler_config.json +19 -0
  19. checkpoints/phendiff/cpg0000/unet/config.json +37 -0
  20. checkpoints/phendiff/cpg0000/unet/diffusion_pytorch_model.bin +3 -0
  21. checkpoints/phendiff/rxrx1/model_index.json +12 -0
  22. checkpoints/phendiff/rxrx1/scheduler/scheduler_config.json +19 -0
  23. checkpoints/phendiff/rxrx1/unet/config.json +37 -0
  24. checkpoints/phendiff/rxrx1/unet/diffusion_pytorch_model.bin +3 -0
checkpoints/cellflux/bbbc021/checkpoint.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a3bf1e8471a6c01565a866faa1f5cf59899165d37dd7d46aeb3ea8552ec59445
3
+ size 899852463
checkpoints/cellflux/cpg0000/checkpoint.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:486b4491ede0c425b7a1f14ea221891c3b66dcbb9820bb2053e6af9b939d0eb1
3
+ size 901638255
checkpoints/cellflux/rxrx1/checkpoint.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fa61925e0e69f5ee49c59518e6d0475c8519e40273e8bd63cfda29cd95e801b7
3
+ size 893323247
checkpoints/impa/bbbc021/embeddings.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:32564991acd6f19fd3d39f41f7a7ff8c5c0e951e6ee3d590ac1a9195127ad292
3
+ size 107854
checkpoints/impa/bbbc021/nets.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9e66e7ee61f5a78afc7433224f66b70ea785f3f1a3b8a143ec6e1a4a0b120bb5
3
+ size 212153202
checkpoints/impa/bbbc021/optims.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dc593b509469f6c5e117b0db3b918336880e8ab55ba0449e8a71590a952a819f
3
+ size 424327786
checkpoints/impa/cpg0000/embeddings.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:001c7682013e8cb72a586f510ccca36b5fb925e0d6760061428fcfdb8ed5008c
3
+ size 3658638
checkpoints/impa/cpg0000/nets.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c629d005cdbb3b3ae44aba8ee8c2eabc924cbde85502229844dea5b45eaef418
3
+ size 213699250
checkpoints/impa/cpg0000/optims.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3b46bc23d2478b1a11b5c5b11454edc4621c7222279bf7013f6cd25576d60fb4
3
+ size 427419882
checkpoints/impa/rxrx1/embeddings.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1a9cc77de53bfd20862e7f19b19c6eb667b9d9d50459b7bf12edf47ef06e5f1f
3
+ size 418126
checkpoints/impa/rxrx1/nets.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bec9a4be2f3650c769a28183ad79877061f9bad6de245b706f2b039e9d2edd36
3
+ size 214021490
checkpoints/impa/rxrx1/optims.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:62ae7c6058633786426da85508d44f5d748020375f58f22a923974a126be582f
3
+ size 428898742
checkpoints/phendiff/bbbc021/model_index.json ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "ConditionalDDIMPipeline",
3
+ "_diffusers_version": "0.18.2",
4
+ "scheduler": [
5
+ "diffusers",
6
+ "DDIMScheduler"
7
+ ],
8
+ "unet": [
9
+ "src.cond_unet_2d.cond_unet_2d",
10
+ "CustomCondUNet2DModel"
11
+ ]
12
+ }
checkpoints/phendiff/bbbc021/scheduler/scheduler_config.json ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "DDIMScheduler",
3
+ "_diffusers_version": "0.18.2",
4
+ "beta_end": 0.02,
5
+ "beta_schedule": "scaled_linear",
6
+ "beta_start": 0.0001,
7
+ "clip_sample": true,
8
+ "clip_sample_range": 1.0,
9
+ "dynamic_thresholding_ratio": 0.995,
10
+ "num_train_timesteps": 3000,
11
+ "prediction_type": "v_prediction",
12
+ "rescale_betas_zero_snr": true,
13
+ "sample_max_value": 1.0,
14
+ "set_alpha_to_one": true,
15
+ "steps_offset": 0,
16
+ "thresholding": false,
17
+ "timestep_spacing": "trailing",
18
+ "trained_betas": null
19
+ }
checkpoints/phendiff/bbbc021/unet/config.json ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "CustomCondUNet2DModel",
3
+ "_diffusers_version": "0.18.2",
4
+ "act_fn": "silu",
5
+ "add_attention": true,
6
+ "attention_head_dim": 8,
7
+ "block_out_channels": [
8
+ 64,
9
+ 128,
10
+ 256
11
+ ],
12
+ "center_input_sample": false,
13
+ "class_embed_type": null,
14
+ "down_block_types": [
15
+ "DownBlock2D",
16
+ "DownBlock2D",
17
+ "AttnDownBlock2D"
18
+ ],
19
+ "downsample_padding": 1,
20
+ "flip_sin_to_cos": true,
21
+ "freq_shift": 0,
22
+ "in_channels": 3,
23
+ "layers_per_block": 2,
24
+ "mid_block_scale_factor": 1,
25
+ "norm_eps": 1e-05,
26
+ "norm_num_groups": 32,
27
+ "num_class_embeds": 26,
28
+ "out_channels": 3,
29
+ "resnet_time_scale_shift": "default",
30
+ "sample_size": 128,
31
+ "time_embedding_type": "positional",
32
+ "up_block_types": [
33
+ "AttnUpBlock2D",
34
+ "UpBlock2D",
35
+ "UpBlock2D"
36
+ ]
37
+ }
checkpoints/phendiff/bbbc021/unet/diffusion_pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ff4988054705aeda1cb9fea5deedbf7271626b2a88ac048240c610e133b0a082
3
+ size 63031748
checkpoints/phendiff/cpg0000/model_index.json ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "ConditionalDDIMPipeline",
3
+ "_diffusers_version": "0.18.2",
4
+ "scheduler": [
5
+ "diffusers",
6
+ "DDIMScheduler"
7
+ ],
8
+ "unet": [
9
+ "src.cond_unet_2d.cond_unet_2d",
10
+ "CustomCondUNet2DModel"
11
+ ]
12
+ }
checkpoints/phendiff/cpg0000/scheduler/scheduler_config.json ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "DDIMScheduler",
3
+ "_diffusers_version": "0.18.2",
4
+ "beta_end": 0.02,
5
+ "beta_schedule": "scaled_linear",
6
+ "beta_start": 0.0001,
7
+ "clip_sample": true,
8
+ "clip_sample_range": 1.0,
9
+ "dynamic_thresholding_ratio": 0.995,
10
+ "num_train_timesteps": 3000,
11
+ "prediction_type": "v_prediction",
12
+ "rescale_betas_zero_snr": true,
13
+ "sample_max_value": 1.0,
14
+ "set_alpha_to_one": true,
15
+ "steps_offset": 0,
16
+ "thresholding": false,
17
+ "timestep_spacing": "trailing",
18
+ "trained_betas": null
19
+ }
checkpoints/phendiff/cpg0000/unet/config.json ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "CustomCondUNet2DModel",
3
+ "_diffusers_version": "0.18.2",
4
+ "act_fn": "silu",
5
+ "add_attention": true,
6
+ "attention_head_dim": 8,
7
+ "block_out_channels": [
8
+ 64,
9
+ 128,
10
+ 256
11
+ ],
12
+ "center_input_sample": false,
13
+ "class_embed_type": null,
14
+ "down_block_types": [
15
+ "DownBlock2D",
16
+ "DownBlock2D",
17
+ "AttnDownBlock2D"
18
+ ],
19
+ "downsample_padding": 1,
20
+ "flip_sin_to_cos": true,
21
+ "freq_shift": 0,
22
+ "in_channels": 5,
23
+ "layers_per_block": 2,
24
+ "mid_block_scale_factor": 1,
25
+ "norm_eps": 1e-05,
26
+ "norm_num_groups": 32,
27
+ "num_class_embeds": 747,
28
+ "out_channels": 5,
29
+ "resnet_time_scale_shift": "default",
30
+ "sample_size": 128,
31
+ "time_embedding_type": "positional",
32
+ "up_block_types": [
33
+ "AttnUpBlock2D",
34
+ "UpBlock2D",
35
+ "UpBlock2D"
36
+ ]
37
+ }
checkpoints/phendiff/cpg0000/unet/diffusion_pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ce569a6f83cca2d6eb335c270f745ac5ce25380608c263dd59390f604b7a2f80
3
+ size 63779268
checkpoints/phendiff/rxrx1/model_index.json ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "ConditionalDDIMPipeline",
3
+ "_diffusers_version": "0.18.2",
4
+ "scheduler": [
5
+ "diffusers",
6
+ "DDIMScheduler"
7
+ ],
8
+ "unet": [
9
+ "src.cond_unet_2d.cond_unet_2d",
10
+ "CustomCondUNet2DModel"
11
+ ]
12
+ }
checkpoints/phendiff/rxrx1/scheduler/scheduler_config.json ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "DDIMScheduler",
3
+ "_diffusers_version": "0.18.2",
4
+ "beta_end": 0.02,
5
+ "beta_schedule": "scaled_linear",
6
+ "beta_start": 0.0001,
7
+ "clip_sample": true,
8
+ "clip_sample_range": 1.0,
9
+ "dynamic_thresholding_ratio": 0.995,
10
+ "num_train_timesteps": 3000,
11
+ "prediction_type": "v_prediction",
12
+ "rescale_betas_zero_snr": true,
13
+ "sample_max_value": 1.0,
14
+ "set_alpha_to_one": true,
15
+ "steps_offset": 0,
16
+ "thresholding": false,
17
+ "timestep_spacing": "trailing",
18
+ "trained_betas": null
19
+ }
checkpoints/phendiff/rxrx1/unet/config.json ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "CustomCondUNet2DModel",
3
+ "_diffusers_version": "0.18.2",
4
+ "act_fn": "silu",
5
+ "add_attention": true,
6
+ "attention_head_dim": 8,
7
+ "block_out_channels": [
8
+ 64,
9
+ 128,
10
+ 256
11
+ ],
12
+ "center_input_sample": false,
13
+ "class_embed_type": null,
14
+ "down_block_types": [
15
+ "DownBlock2D",
16
+ "DownBlock2D",
17
+ "AttnDownBlock2D"
18
+ ],
19
+ "downsample_padding": 1,
20
+ "flip_sin_to_cos": true,
21
+ "freq_shift": 0,
22
+ "in_channels": 6,
23
+ "layers_per_block": 2,
24
+ "mid_block_scale_factor": 1,
25
+ "norm_eps": 1e-05,
26
+ "norm_num_groups": 32,
27
+ "num_class_embeds": 1042,
28
+ "out_channels": 6,
29
+ "resnet_time_scale_shift": "default",
30
+ "sample_size": 128,
31
+ "time_embedding_type": "positional",
32
+ "up_block_types": [
33
+ "AttnUpBlock2D",
34
+ "UpBlock2D",
35
+ "UpBlock2D"
36
+ ]
37
+ }
checkpoints/phendiff/rxrx1/unet/diffusion_pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:176605e488796d301e1d773ca12995ab66a2e9f7a03f7c48e20d12d736b6a809
3
+ size 64085956