Patrick Haller
commited on
Commit
·
cca7565
1
Parent(s):
b9037b6
Fix
Browse files- modeling_xlstm.py +1 -0
modeling_xlstm.py
CHANGED
|
@@ -219,6 +219,7 @@ class xLSTMForSequenceClassification(xLSTMPreTrainedModel):
|
|
| 219 |
|
| 220 |
def __init__(self, config: xLSTMConfig, **kwargs):
|
| 221 |
super().__init__(config)
|
|
|
|
| 222 |
self.config = config
|
| 223 |
self.model = xLSTMModel(config)
|
| 224 |
self.classifier = nn.Linear(config.embedding_dim, config.num_labels, bias=False)
|
|
|
|
| 219 |
|
| 220 |
def __init__(self, config: xLSTMConfig, **kwargs):
|
| 221 |
super().__init__(config)
|
| 222 |
+
self.num_labels = config.num_labels
|
| 223 |
self.config = config
|
| 224 |
self.model = xLSTMModel(config)
|
| 225 |
self.classifier = nn.Linear(config.embedding_dim, config.num_labels, bias=False)
|