shpotes commited on
Commit
d5975be
1 Parent(s): aa9f7af
Files changed (2) hide show
  1. config.json +3 -2
  2. pytorch_model.bin +2 -2
config.json CHANGED
@@ -2,12 +2,13 @@
2
  "_name_or_path": "codegen-350M-mono/pytorch_model.bin",
3
  "activation_function": "gelu_new",
4
  "architectures": [
5
- "CodeGenModel"
6
  ],
7
  "attn_pdrop": 0.0,
8
  "auto_map": {
9
  "AutoConfig": "configuration_codegen.CodeGenConfig",
10
- "AutoModel": "modelling_codegen.CodeGenModel"
 
11
  },
12
  "bos_token_id": 1,
13
  "embd_pdrop": 0.0,
 
2
  "_name_or_path": "codegen-350M-mono/pytorch_model.bin",
3
  "activation_function": "gelu_new",
4
  "architectures": [
5
+ "CodeGenForCausalLM"
6
  ],
7
  "attn_pdrop": 0.0,
8
  "auto_map": {
9
  "AutoConfig": "configuration_codegen.CodeGenConfig",
10
+ "AutoModel": "modelling_codegen.CodeGenModel",
11
+ "AutoModelForCausalLM": "modelling_codegen.CodeGenForCausalLM"
12
  },
13
  "bos_token_id": 1,
14
  "embd_pdrop": 0.0,
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7716cb85d91befd8ac38792fe7d78a0b6aec15bd07057946a13e044323a6e2b0
3
- size 1300884331
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:da737dea7f1960dbac5e72975155731ac41294af8b7076daf14d5af86da20889
3
+ size 1510810343