Update modeling.py
Browse files- modeling.py +5 -5
modeling.py
CHANGED
|
@@ -41,11 +41,11 @@ class RobertaEmoPillars(PreTrainedModel):
|
|
| 41 |
self.post_init() # ensures HF weights init
|
| 42 |
|
| 43 |
def forward(self, input_ids=None, attention_mask=None, **kwargs):
|
| 44 |
-
|
| 45 |
-
|
| 46 |
-
|
| 47 |
-
|
| 48 |
-
|
| 49 |
|
| 50 |
|
| 51 |
# --- Registration for AutoModel ---
|
|
|
|
| 41 |
self.post_init() # ensures HF weights init
|
| 42 |
|
| 43 |
def forward(self, input_ids=None, attention_mask=None, **kwargs):
|
| 44 |
+
inputs = {
|
| 45 |
+
"input_ids": input_ids,
|
| 46 |
+
"attention_mask": attention_mask,
|
| 47 |
+
**kwargs
|
| 48 |
+
}
|
| 49 |
|
| 50 |
|
| 51 |
# --- Registration for AutoModel ---
|