isacat commited on
Commit
bb35eba
1 Parent(s): 6dd94b7

feat: make automodel load from t5encodermodel

Browse files
Files changed (2) hide show
  1. config.json +4 -1
  2. modeling_t5.py +1 -0
config.json CHANGED
@@ -3,6 +3,9 @@
3
  "architectures": [
4
  "T5EncoderModel"
5
  ],
 
 
 
6
  "d_ff": 2048,
7
  "d_kv": 64,
8
  "d_model": 512,
@@ -49,4 +52,4 @@
49
  }
50
  },
51
  "vocab_size": 32128
52
- }
 
3
  "architectures": [
4
  "T5EncoderModel"
5
  ],
6
+ "auto_map": {
7
+ "AutoModel": "modeling_t5.T5EncoderModel"
8
+ },
9
  "d_ff": 2048,
10
  "d_kv": 64,
11
  "d_model": 512,
 
52
  }
53
  },
54
  "vocab_size": 32128
55
+ }
modeling_t5.py ADDED
@@ -0,0 +1 @@
 
 
1
+ from transformers.models.t5.modeling_t5 import T5EncoderModel