vittoriopippi
commited on
Commit
·
385308e
1
Parent(s):
0f0df26
Error with model_type
Browse files- modeling_vatrpp.py +1 -1
modeling_vatrpp.py
CHANGED
@@ -132,5 +132,5 @@ class VATrPP(PreTrainedModel):
|
|
132 |
return result
|
133 |
|
134 |
|
135 |
-
AutoConfig.
|
136 |
AutoModel.register(VATrPPConfig, VATrPP)
|
|
|
132 |
return result
|
133 |
|
134 |
|
135 |
+
AutoConfig.register("vatrpp", VATrPPConfig)
|
136 |
AutoModel.register(VATrPPConfig, VATrPP)
|