aw777 commited on
Commit
3c3e10b
·
verified ·
1 Parent(s): 3cfed70

Add model configuration for download tracking

Browse files
Files changed (1) hide show
  1. config.json +18 -0
config.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "kumo_tabular",
3
+ "default_size": "large",
4
+ "checkpoint_files": {
5
+ "small": {
6
+ "classification": "small/classifier.pt",
7
+ "regression": "small/regressor.pt"
8
+ },
9
+ "medium": {
10
+ "classification": "medium/classifier.pt",
11
+ "regression": "medium/regressor.pt"
12
+ },
13
+ "large": {
14
+ "classification": "large/classifier.pt",
15
+ "regression": "large/regressor.pt"
16
+ }
17
+ }
18
+ }