Jiqing commited on
Commit
9160547
·
1 Parent(s): 22c9eb3

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +2 -1
config.json CHANGED
@@ -3,7 +3,8 @@
3
  "ProtSTModel"
4
  ],
5
  "auto_map": {
6
- "AutoModel": "modeling_protst.ProtSTModel"
 
7
  },
8
  "protein_config": {
9
  "_name_or_path": "/tmp/facebook/esm1b_t33_650M_UR50S",
 
3
  "ProtSTModel"
4
  ],
5
  "auto_map": {
6
+ "AutoModel": "modeling_protst.ProtSTModel",
7
+ "AutoConfig": "configuration_protst.ProtSTConfig"
8
  },
9
  "protein_config": {
10
  "_name_or_path": "/tmp/facebook/esm1b_t33_650M_UR50S",