Update flaubert2_model.py
Browse files- flaubert2_model.py +2 -2
flaubert2_model.py
CHANGED
@@ -259,9 +259,9 @@ class Flaubert2Encoder(RobertaEncoder):
|
|
259 |
|
260 |
|
261 |
class Flaubert2Model(RobertaModel):
|
262 |
-
|
263 |
def __init__(self, config, **kwargs):
|
264 |
-
|
265 |
base_model_prefix = "flaubert2"
|
266 |
|
267 |
super().__init__(config, **kwargs)
|
|
|
259 |
|
260 |
|
261 |
class Flaubert2Model(RobertaModel):
|
262 |
+
config_class = Flaubert2Config
|
263 |
def __init__(self, config, **kwargs):
|
264 |
+
|
265 |
base_model_prefix = "flaubert2"
|
266 |
|
267 |
super().__init__(config, **kwargs)
|