Hugging Face's logo

google
/
tabfm-1.0.0-pytorch

Tabular Classification
Safetensors
PyTorch
tabfm
tabular
tabular-regression
zero-shot
in-context-learning
foundation-model
Model card Files
xet
 Community
7

convert weights to safetensors

#3
by kashif HF Staff - opened
base: refs/heads/main
←
from: refs/pr/3
Discussion Files changed
+30
-32
Files changed (4) hide show
  1. classification/config.json +12 -13
  2. classification/{pytorch_model.bin → model.safetensors} +2 -2
  3. regression/config.json +12 -13
  4. regression/{pytorch_model.bin → model.safetensors} +2 -2
classification/config.json CHANGED
@@ -1,18 +1,17 @@
1
  {
2
- "model_type": "tabfm",
3
- "version": "1.0.0",
4
- "task": "classification",
5
- "framework": "pytorch",
6
- "embed_dim": 256,
7
- "max_classes": 10,
8
- "col_num_blocks": 3,
9
  "col_nhead": 4,
 
10
  "col_num_inds": 256,
11
- "row_num_blocks": 3,
12
- "row_nhead": 8,
13
- "row_num_cls": 8,
14
- "icl_num_blocks": 24,
15
- "icl_nhead": 8,
16
  "ff_factor": 4,
17
- "feature_group_size": 3
 
 
 
 
 
 
 
18
  }
 
1
  {
 
 
 
 
 
 
 
2
  "col_nhead": 4,
3
+ "col_num_blocks": 3,
4
  "col_num_inds": 256,
5
+ "decoder_hidden": null,
6
+ "embed_dim": 256,
7
+ "feature_group_size": 3,
 
 
8
  "ff_factor": 4,
9
+ "icl_nhead": 8,
10
+ "icl_num_blocks": 24,
11
+ "is_classifier": true,
12
+ "max_classes": 10,
13
+ "num_freq": 32,
14
+ "row_nhead": 8,
15
+ "row_num_blocks": 3,
16
+ "row_num_cls": 8
17
  }
classification/{pytorch_model.bin → model.safetensors} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:51bf5aff974c3ee499691f6dd4d38b4b485b55964acd889dfcb959a8eaa66778
3
- size 6558126047
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:928cb350becdc77cdb7a9e8c36deda88917bfd14a3091894a2dc516db58a2085
3
+ size 6557888408
regression/config.json CHANGED
@@ -1,18 +1,17 @@
1
  {
2
- "model_type": "tabfm",
3
- "version": "1.0.0",
4
- "task": "regression",
5
- "framework": "pytorch",
6
- "embed_dim": 256,
7
- "max_classes": 10,
8
- "col_num_blocks": 3,
9
  "col_nhead": 4,
 
10
  "col_num_inds": 256,
11
- "row_num_blocks": 3,
12
- "row_nhead": 8,
13
- "row_num_cls": 8,
14
- "icl_num_blocks": 24,
15
- "icl_nhead": 8,
16
  "ff_factor": 4,
17
- "feature_group_size": 3
 
 
 
 
 
 
 
18
  }
 
1
  {
 
 
 
 
 
 
 
2
  "col_nhead": 4,
3
+ "col_num_blocks": 3,
4
  "col_num_inds": 256,
5
+ "decoder_hidden": null,
6
+ "embed_dim": 256,
7
+ "feature_group_size": 3,
 
 
8
  "ff_factor": 4,
9
+ "icl_nhead": 8,
10
+ "icl_num_blocks": 24,
11
+ "is_classifier": false,
12
+ "max_classes": 10,
13
+ "num_freq": 32,
14
+ "row_nhead": 8,
15
+ "row_num_blocks": 3,
16
+ "row_num_cls": 8
17
  }
regression/{pytorch_model.bin → model.safetensors} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:88bd23418fcd9f685afa87460f9b68d03933450be453567d3c055307ab15a364
3
- size 6591482635
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bd5a615b0322a8f04a895038de6df6fbd71430eca750e1d792f31048654674a9
3
+ size 6591243724