Update modelling_hat.py
Browse files- modelling_hat.py +1 -1
modelling_hat.py
CHANGED
@@ -381,7 +381,7 @@ class HATConfig(PretrainedConfig):
|
|
381 |
classifier_dropout (:obj:`float`, `optional`):
|
382 |
The dropout ratio for the classification head.
|
383 |
"""
|
384 |
-
model_type = "
|
385 |
|
386 |
def __init__(
|
387 |
self,
|
|
|
381 |
classifier_dropout (:obj:`float`, `optional`):
|
382 |
The dropout ratio for the classification head.
|
383 |
"""
|
384 |
+
model_type = "hierarchical-transformer"
|
385 |
|
386 |
def __init__(
|
387 |
self,
|