Hugging Face's logo

Goodeat
/
drifting

Model card Files
xet
 Community
Goodeat commited on
Commit
1a5afa9
·
0 Parent(s):

initial commit

Browse files
Files changed (17) hide show
  1. .gitattributes +35 -0
  2. models/gen/jax/ablation/ema_params.msgpack +3 -0
  3. models/gen/jax/ablation/metadata.json +36 -0
  4. models/gen/jax/latent_B_sota/ema_params.msgpack +3 -0
  5. models/gen/jax/latent_B_sota/metadata.json +36 -0
  6. models/gen/jax/latent_L_sota/ema_params.msgpack +3 -0
  7. models/gen/jax/latent_L_sota/metadata.json +36 -0
  8. models/gen/jax/pixel_B_sota/ema_params.msgpack +3 -0
  9. models/gen/jax/pixel_B_sota/metadata.json +37 -0
  10. models/gen/jax/pixel_L_sota/ema_params.msgpack +3 -0
  11. models/gen/jax/pixel_L_sota/metadata.json +37 -0
  12. models/mae/jax/mae_latent_256/ema_params.msgpack +3 -0
  13. models/mae/jax/mae_latent_256/metadata.json +29 -0
  14. models/mae/jax/mae_latent_640/ema_params.msgpack +3 -0
  15. models/mae/jax/mae_latent_640/metadata.json +29 -0
  16. models/mae/jax/mae_pixel_640/ema_params.msgpack +3 -0
  17. models/mae/jax/mae_pixel_640/metadata.json +29 -0
.gitattributes ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ *.7z filter=lfs diff=lfs merge=lfs -text
2
+ *.arrow filter=lfs diff=lfs merge=lfs -text
3
+ *.bin filter=lfs diff=lfs merge=lfs -text
4
+ *.bz2 filter=lfs diff=lfs merge=lfs -text
5
+ *.ckpt filter=lfs diff=lfs merge=lfs -text
6
+ *.ftz filter=lfs diff=lfs merge=lfs -text
7
+ *.gz filter=lfs diff=lfs merge=lfs -text
8
+ *.h5 filter=lfs diff=lfs merge=lfs -text
9
+ *.joblib filter=lfs diff=lfs merge=lfs -text
10
+ *.lfs.* filter=lfs diff=lfs merge=lfs -text
11
+ *.mlmodel filter=lfs diff=lfs merge=lfs -text
12
+ *.model filter=lfs diff=lfs merge=lfs -text
13
+ *.msgpack filter=lfs diff=lfs merge=lfs -text
14
+ *.npy filter=lfs diff=lfs merge=lfs -text
15
+ *.npz filter=lfs diff=lfs merge=lfs -text
16
+ *.onnx filter=lfs diff=lfs merge=lfs -text
17
+ *.ot filter=lfs diff=lfs merge=lfs -text
18
+ *.parquet filter=lfs diff=lfs merge=lfs -text
19
+ *.pb filter=lfs diff=lfs merge=lfs -text
20
+ *.pickle filter=lfs diff=lfs merge=lfs -text
21
+ *.pkl filter=lfs diff=lfs merge=lfs -text
22
+ *.pt filter=lfs diff=lfs merge=lfs -text
23
+ *.pth filter=lfs diff=lfs merge=lfs -text
24
+ *.rar filter=lfs diff=lfs merge=lfs -text
25
+ *.safetensors filter=lfs diff=lfs merge=lfs -text
26
+ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
27
+ *.tar.* filter=lfs diff=lfs merge=lfs -text
28
+ *.tar filter=lfs diff=lfs merge=lfs -text
29
+ *.tflite filter=lfs diff=lfs merge=lfs -text
30
+ *.tgz filter=lfs diff=lfs merge=lfs -text
31
+ *.wasm filter=lfs diff=lfs merge=lfs -text
32
+ *.xz filter=lfs diff=lfs merge=lfs -text
33
+ *.zip filter=lfs diff=lfs merge=lfs -text
34
+ *.zst filter=lfs diff=lfs merge=lfs -text
35
+ *tfevents* filter=lfs diff=lfs merge=lfs -text
models/gen/jax/ablation/ema_params.msgpack ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b1a669358e52319eb021485482d57e65e63627fcb4a86153dc3a809f161d09da
3
+ size 530846017
models/gen/jax/ablation/metadata.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schema_version": 1,
3
+ "artifact_type": "ema_model",
4
+ "kind": "gen",
5
+ "backend": "jax",
6
+ "model_id": "ablation",
7
+ "config_name": "ablation",
8
+ "config_path": "configs/gen/latent_ablation.yaml",
9
+ "model_config": {
10
+ "cond_dim": 768,
11
+ "input_size": 32,
12
+ "in_channels": 4,
13
+ "patch_size": 2,
14
+ "hidden_size": 768,
15
+ "depth": 12,
16
+ "num_heads": 12,
17
+ "mlp_ratio": 4.0,
18
+ "out_channels": 4,
19
+ "use_qknorm": true,
20
+ "use_swiglu": true,
21
+ "use_rope": true,
22
+ "use_rmsnorm": true,
23
+ "n_cls_tokens": 16,
24
+ "noise_classes": 64,
25
+ "noise_coords": 32,
26
+ "use_bf16": true,
27
+ "attn_fp32": true,
28
+ "num_classes": 1000
29
+ },
30
+ "source": {
31
+ "step_requested": "30000",
32
+ "step_loaded": 30000,
33
+ "ema_requested": 0.999,
34
+ "ema_selected": "0.999"
35
+ }
36
+ }
models/gen/jax/latent_B_sota/ema_params.msgpack ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ad1505125d09316c2cda15e7294b49be2ab7919ce16fab1caee0daa467ea402b
3
+ size 530846017
models/gen/jax/latent_B_sota/metadata.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schema_version": 1,
3
+ "artifact_type": "ema_model",
4
+ "kind": "gen",
5
+ "backend": "jax",
6
+ "model_id": "latent_B_sota",
7
+ "config_name": "latent_B_sota",
8
+ "config_path": "configs/gen/latent_sota_B.yaml",
9
+ "model_config": {
10
+ "cond_dim": 768,
11
+ "input_size": 32,
12
+ "in_channels": 4,
13
+ "patch_size": 2,
14
+ "hidden_size": 768,
15
+ "depth": 12,
16
+ "num_heads": 12,
17
+ "mlp_ratio": 4.0,
18
+ "out_channels": 4,
19
+ "use_qknorm": true,
20
+ "use_swiglu": true,
21
+ "use_rope": true,
22
+ "use_rmsnorm": true,
23
+ "n_cls_tokens": 16,
24
+ "noise_classes": 64,
25
+ "noise_coords": 32,
26
+ "use_bf16": true,
27
+ "attn_fp32": false,
28
+ "num_classes": 1000
29
+ },
30
+ "source": {
31
+ "step_requested": "200000",
32
+ "step_loaded": 200000,
33
+ "ema_requested": 0.9999,
34
+ "ema_selected": "0.9999"
35
+ }
36
+ }
models/gen/jax/latent_L_sota/ema_params.msgpack ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:622763e1fbccabdeba7d99a6b5568fa5ab3b1617221ca39295d7801466cab332
3
+ size 1851656457
models/gen/jax/latent_L_sota/metadata.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schema_version": 1,
3
+ "artifact_type": "ema_model",
4
+ "kind": "gen",
5
+ "backend": "jax",
6
+ "model_id": "latent_L_sota",
7
+ "config_name": "latent_L_sota",
8
+ "config_path": "configs/gen/latent_sota_L.yaml",
9
+ "model_config": {
10
+ "cond_dim": 1024,
11
+ "input_size": 32,
12
+ "in_channels": 4,
13
+ "patch_size": 2,
14
+ "hidden_size": 1024,
15
+ "depth": 24,
16
+ "num_heads": 16,
17
+ "mlp_ratio": 4.0,
18
+ "out_channels": 4,
19
+ "use_qknorm": true,
20
+ "use_swiglu": true,
21
+ "use_rope": true,
22
+ "use_rmsnorm": true,
23
+ "n_cls_tokens": 16,
24
+ "noise_classes": 64,
25
+ "noise_coords": 32,
26
+ "use_bf16": true,
27
+ "attn_fp32": false,
28
+ "num_classes": 1000
29
+ },
30
+ "source": {
31
+ "step_requested": "200000",
32
+ "step_loaded": 200000,
33
+ "ema_requested": 0.9999,
34
+ "ema_selected": "0.9999"
35
+ }
36
+ }
models/gen/jax/pixel_B_sota/ema_params.msgpack ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d0de0b03f9907b7da872e96bfb7be29002a4833cd35dd59c696b57ad04c51e50
3
+ size 535469329
models/gen/jax/pixel_B_sota/metadata.json ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schema_version": 1,
3
+ "artifact_type": "ema_model",
4
+ "kind": "gen",
5
+ "backend": "jax",
6
+ "model_id": "pixel_B_sota",
7
+ "config_name": "pixel_B_sota",
8
+ "config_path": "configs/gen/pixel_sota_B.yaml",
9
+ "model_config": {
10
+ "cond_dim": 768,
11
+ "input_size": 256,
12
+ "in_channels": 3,
13
+ "patch_size": 16,
14
+ "hidden_size": 768,
15
+ "depth": 12,
16
+ "num_heads": 12,
17
+ "mlp_ratio": 4.0,
18
+ "out_channels": 3,
19
+ "use_qknorm": true,
20
+ "use_swiglu": true,
21
+ "use_rope": true,
22
+ "use_rmsnorm": true,
23
+ "n_cls_tokens": 16,
24
+ "noise_classes": 64,
25
+ "noise_coords": 32,
26
+ "use_bf16": true,
27
+ "attn_fp32": true,
28
+ "use_remat": true,
29
+ "num_classes": 1000
30
+ },
31
+ "source": {
32
+ "step_requested": "100000",
33
+ "step_loaded": 100000,
34
+ "ema_requested": 0.999,
35
+ "ema_selected": "0.999"
36
+ }
37
+ }
models/gen/jax/pixel_L_sota/ema_params.msgpack ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5b042c036577404ab09502425e96f2740b724b846d3db6a1a8080e3fd6f91bdc
3
+ size 1857819857
models/gen/jax/pixel_L_sota/metadata.json ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schema_version": 1,
3
+ "artifact_type": "ema_model",
4
+ "kind": "gen",
5
+ "backend": "jax",
6
+ "model_id": "pixel_L_sota",
7
+ "config_name": "pixel_L_sota",
8
+ "config_path": "configs/gen/pixel_sota_L.yaml",
9
+ "model_config": {
10
+ "cond_dim": 1024,
11
+ "input_size": 256,
12
+ "in_channels": 3,
13
+ "patch_size": 16,
14
+ "hidden_size": 1024,
15
+ "depth": 24,
16
+ "num_heads": 16,
17
+ "mlp_ratio": 4.0,
18
+ "out_channels": 3,
19
+ "use_qknorm": true,
20
+ "use_swiglu": true,
21
+ "use_rope": true,
22
+ "use_rmsnorm": true,
23
+ "n_cls_tokens": 16,
24
+ "noise_classes": 64,
25
+ "noise_coords": 32,
26
+ "use_bf16": true,
27
+ "attn_fp32": true,
28
+ "use_remat": true,
29
+ "num_classes": 1000
30
+ },
31
+ "source": {
32
+ "step_requested": "100000",
33
+ "step_loaded": 100000,
34
+ "ema_requested": 0.999,
35
+ "ema_selected": "0.999"
36
+ }
37
+ }
models/mae/jax/mae_latent_256/ema_params.msgpack ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ecaafa5e4233567916b479a94842a853392a72c88e3535769e1081110824944f
3
+ size 1725422798
models/mae/jax/mae_latent_256/metadata.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schema_version": 1,
3
+ "artifact_type": "ema_model",
4
+ "kind": "mae",
5
+ "backend": "jax",
6
+ "model_id": "mae_latent_256",
7
+ "config_name": "mae_latent_256",
8
+ "config_path": "configs/gen/latent_ablation.yaml",
9
+ "model_config": {
10
+ "base_channels": 256,
11
+ "patch_size": 2,
12
+ "dropout_prob": 0.0,
13
+ "layers": [
14
+ 3,
15
+ 4,
16
+ 6,
17
+ 3
18
+ ],
19
+ "in_channels": 4,
20
+ "use_bf16": true,
21
+ "input_patch_size": 1
22
+ },
23
+ "source": {
24
+ "step_requested": "30000",
25
+ "step_loaded": 30000,
26
+ "ema_requested": 0.9995,
27
+ "ema_selected": "0.9995"
28
+ }
29
+ }
models/mae/jax/mae_latent_640/ema_params.msgpack ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dbac36fdad0cd572df7fbbeea7922b55348dfa5be56a2ccb8ca4a90d592d75b6
3
+ size 10751471826
models/mae/jax/mae_latent_640/metadata.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schema_version": 1,
3
+ "artifact_type": "ema_model",
4
+ "kind": "mae",
5
+ "backend": "jax",
6
+ "model_id": "mae_latent_640",
7
+ "config_name": "mae_latent_640",
8
+ "config_path": "configs/gen/latent_sota_B.yaml,configs/gen/latent_sota_L.yaml",
9
+ "model_config": {
10
+ "base_channels": 640,
11
+ "patch_size": 2,
12
+ "dropout_prob": 0.0,
13
+ "layers": [
14
+ 3,
15
+ 4,
16
+ 6,
17
+ 3
18
+ ],
19
+ "in_channels": 4,
20
+ "use_bf16": true,
21
+ "input_patch_size": 1
22
+ },
23
+ "source": {
24
+ "step_requested": "200000",
25
+ "step_loaded": 200000,
26
+ "ema_requested": 0.9995,
27
+ "ema_selected": "0.9995"
28
+ }
29
+ }
models/mae/jax/mae_pixel_640/ema_params.msgpack ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1d68c26b1b491888c028fcdbcd6e03aab14ab3aa9bc8f6be68adf22d3a86125c
3
+ size 10756285387
models/mae/jax/mae_pixel_640/metadata.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schema_version": 1,
3
+ "artifact_type": "ema_model",
4
+ "kind": "mae",
5
+ "backend": "jax",
6
+ "model_id": "mae_pixel_640",
7
+ "config_name": "mae_pixel_640",
8
+ "config_path": "configs/gen/pixel_sota_B.yaml,configs/gen/pixel_sota_L.yaml",
9
+ "model_config": {
10
+ "base_channels": 640,
11
+ "patch_size": 2,
12
+ "dropout_prob": 0.0,
13
+ "layers": [
14
+ 3,
15
+ 4,
16
+ 6,
17
+ 3
18
+ ],
19
+ "in_channels": 3,
20
+ "use_bf16": true,
21
+ "input_patch_size": 8
22
+ },
23
+ "source": {
24
+ "step_requested": "200000",
25
+ "step_loaded": 200000,
26
+ "ema_requested": 0.9995,
27
+ "ema_selected": "0.9995"
28
+ }
29
+ }