stefan-it commited on
Commit
9d21367
1 Parent(s): 7b7eb08

config: add mapping for AutoModelForSequenceClassification to own xLSTMForSequenceClassification

Browse files
Files changed (1) hide show
  1. config.json +1 -0
config.json CHANGED
@@ -24,6 +24,7 @@
24
  "auto_map": {
25
  "AutoConfig": "configuration_xlstm.xLSTMConfig",
26
  "AutoModelForCausalLM": "modeling_xlstm.xLSTMForCausalLM",
 
27
  "AutoModel": "modeling_xlstm.xLSTMModel"
28
  }
29
  }
 
24
  "auto_map": {
25
  "AutoConfig": "configuration_xlstm.xLSTMConfig",
26
  "AutoModelForCausalLM": "modeling_xlstm.xLSTMForCausalLM",
27
+ "AutoModelForSequenceClassification": "modeling_xlstm.xLSTMForSequenceClassification",
28
  "AutoModel": "modeling_xlstm.xLSTMModel"
29
  }
30
  }