mmcarpi commited on
Commit
bebeceb
·
verified ·
1 Parent(s): a9c0b22

Upload custom model with source code and tokenizer

Browse files
Files changed (1) hide show
  1. config.json +4 -1
config.json CHANGED
@@ -3,7 +3,10 @@
3
  "FlexQwenForCausalLM"
4
  ],
5
  "auto_map": {
6
- "AutoModelForCausalLM": "qwen.FlexQwenForCausalLM"
 
 
 
7
  },
8
  "cls_token_id": 1,
9
  "context_length": 4096,
 
3
  "FlexQwenForCausalLM"
4
  ],
5
  "auto_map": {
6
+ "AutoConfig": "qwen.FlexQwenConfig",
7
+ "AutoModel": "qwen.FlexQwen",
8
+ "AutoModelForCausalLM": "qwen.FlexQwenForCausalLM",
9
+ "AutoModelForSequenceClassification": "qwen.FlexQwenForSequenceClassification"
10
  },
11
  "cls_token_id": 1,
12
  "context_length": 4096,