igorktech
/

hier-bert-i3-mlm

hierarchical-bert

Generated from Trainer

Model card Files Files and versions Community

igorktech commited on Apr 24, 2024

Commit

787b7b5

·

verified ·

1 Parent(s): 77b797c

Update modelling_hier.py

Files changed (1) hide show

modelling_hier.py +1 -1

modelling_hier.py CHANGED Viewed

@@ -395,7 +395,7 @@ class HierBert(Module):
 class HierBertModel(PreTrainedModel):
     config_class = HierBertConfig
-    base_model_prefix = "hier"
     def __init__(self, config):
         super().__init__(config)

 class HierBertModel(PreTrainedModel):
     config_class = HierBertConfig
+    base_model_prefix = "bert"
     def __init__(self, config):
         super().__init__(config)