igorktech
/

hierarchical-dialog-bert

hierarchical-transformer

Model card Files Files and versions

Metrics Training metrics Community

igorktech commited on Jun 6, 2023

Commit

7f8c0c4

·

1 Parent(s): f199e75

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 = "hi-transformer"
     def __init__(
         self,

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