igorktech
/

hierarchical-dialog-bert

hierarchical-transformer

Model card Files Files and versions Metrics Training metrics Community

igorktech commited on Jun 6, 2023

Commit

65d5357

•

1 Parent(s): 25aee9f

Update configuration_hat.py

Files changed (1) hide show

configuration_hat.py +1 -1

configuration_hat.py CHANGED Viewed

@@ -90,7 +90,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,