Update modeling.py
Browse files- modeling.py +5 -5
modeling.py
CHANGED
|
@@ -50,8 +50,8 @@ class EmoAxis(PreTrainedModel):
|
|
| 50 |
return logits
|
| 51 |
|
| 52 |
|
| 53 |
-
#
|
| 54 |
-
|
| 55 |
-
|
| 56 |
-
|
| 57 |
-
|
|
|
|
| 50 |
return logits
|
| 51 |
|
| 52 |
|
| 53 |
+
# --- Registration for AutoModel ---
|
| 54 |
+
try:
|
| 55 |
+
AutoModel.register(AutoConfig, EmoAxis)
|
| 56 |
+
except:
|
| 57 |
+
pass
|