igorktech
/

hierarchical-dialog-bert

hierarchical-transformer

Model card Files Files and versions Metrics Training metrics Community

igorktech commited on Jun 6, 2023

Commit

25aee9f

•

1 Parent(s): 0e5d19e

Update modelling_hat.py

Files changed (1) hide show

modelling_hat.py +1 -1

modelling_hat.py CHANGED Viewed

@@ -381,7 +381,7 @@ class HATConfig(PretrainedConfig):
         classifier_dropout (:obj:`float`, `optional`):
             The dropout ratio for the classification head.
     """
-    model_type = "hierarchical-transformer"
     def __init__(
         self,

         classifier_dropout (:obj:`float`, `optional`):
             The dropout ratio for the classification head.
     """
+    model_type = "hi-transformer"
     def __init__(
         self,