vittoriopippi
commited on
Commit
·
e3b6daa
1
Parent(s):
e7e85b9
Error with model_type
Browse files- modeling_vatrpp.py +5 -1
modeling_vatrpp.py
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
from transformers import PreTrainedModel, AutoModel
|
| 2 |
from .configuration_vatrpp import VATrPPConfig
|
| 3 |
import json
|
| 4 |
import os
|
|
@@ -130,3 +130,7 @@ class VATrPP(PreTrainedModel):
|
|
| 130 |
x_pos += word.shape[1] + gap_width
|
| 131 |
|
| 132 |
return result
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from transformers import PreTrainedModel, AutoModel, AutoConfig
|
| 2 |
from .configuration_vatrpp import VATrPPConfig
|
| 3 |
import json
|
| 4 |
import os
|
|
|
|
| 130 |
x_pos += word.shape[1] + gap_width
|
| 131 |
|
| 132 |
return result
|
| 133 |
+
|
| 134 |
+
|
| 135 |
+
AutoConfig.register_config("vatrpp", VATrPPConfig)
|
| 136 |
+
AutoModel.register(VATrPPConfig, VATrPP)
|