MDaytek commited on
Commit
e12e271
·
verified ·
1 Parent(s): 10d2fcb

Submission with auto_map and code files

Browse files
Files changed (2) hide show
  1. README.md +1 -0
  2. config.json +4 -0
README.md CHANGED
@@ -9,3 +9,4 @@ license: mit
9
  # chess-v2-head
10
  Model submitted by MDaytek.
11
  **Parameters:** 999,936
 
 
9
  # chess-v2-head
10
  Model submitted by MDaytek.
11
  **Parameters:** 999,936
12
+ **Architecture:** Custom Chess Transformer (Code included)
config.json CHANGED
@@ -2,6 +2,10 @@
2
  "architectures": [
3
  "ChessForCausalLM"
4
  ],
 
 
 
 
5
  "dtype": "float32",
6
  "hidden_size": 128,
7
  "model_type": "chess_transformer",
 
2
  "architectures": [
3
  "ChessForCausalLM"
4
  ],
5
+ "auto_map": {
6
+ "AutoConfig": "model.ChessConfig",
7
+ "AutoModelForCausalLM": "model.ChessForCausalLM"
8
+ },
9
  "dtype": "float32",
10
  "hidden_size": 128,
11
  "model_type": "chess_transformer",