haowei commited on
Commit
cb6f926
1 Parent(s): 932302a
Files changed (2) hide show
  1. config.json +4 -1
  2. roberta_modeling.py +1 -2
config.json CHANGED
@@ -1,6 +1,9 @@
1
  {
2
- "_name_or_path": "/home/haowei/haowei/continual-post-training/agnews_unsup_roberta",
3
  "adapter_mode": "parallel",
 
 
 
 
4
  "adapter_task": 5,
5
  "adapters": {
6
  "adapters": {},
 
1
  {
 
2
  "adapter_mode": "parallel",
3
+ "auto_map": {
4
+ "AutoModelForSequenceClassification": "roberta_modeling.RobertaMaskForSequenceClassification",
5
+ "AutoModelForMaskedLM": "roberta_modeling.RobertaMaskForMaskedLM"
6
+ },
7
  "adapter_task": 5,
8
  "adapters": {
9
  "adapters": {},
roberta_modeling.py CHANGED
@@ -2191,5 +2191,4 @@ class RobertaMaskForSequenceClassification(RobertaMaskBasedModel, RobertaForSequ
2191
  attn_adapter_size=config.attn_adapter_size,
2192
  ffn_adapter_size=config.ffn_adapter_size,
2193
  )
2194
- self.roberta = add_roberta_adapters(self.roberta, adapter_config)
2195
-
 
2191
  attn_adapter_size=config.attn_adapter_size,
2192
  ffn_adapter_size=config.ffn_adapter_size,
2193
  )
2194
+ self.roberta = add_roberta_adapters(self.roberta, adapter_config)