Merge pull request #58 from georgian-io/akash/bugfix

Fix(multimodal_transformers/model/tabular_transformers.py): hidden_dr…
georgian-io · Nov 14, 2023 · 02f4b62 · 02f4b62
2 parents 9ae2eab + c9e49b3
commit 02f4b62
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/multimodal_transformers/model/tabular_transformers.py b/multimodal_transformers/model/tabular_transformers.py
@@ -492,7 +492,7 @@ def __init__(self, hf_model_config):
             self.config.tabular_config = tabular_config.__dict__
 
         tabular_config.text_feat_dim = hf_model_config.hidden_size
-        tabular_config.hidden_dropout_prob = hf_model_config.hidden_dropout_prob
+        tabular_config.hidden_dropout_prob = hf_model_config.dropout
         self.tabular_combiner = TabularFeatCombiner(tabular_config)
         self.num_labels = tabular_config.num_labels
         combined_feat_dim = self.tabular_combiner.final_out_dim
@@ -604,7 +604,7 @@ def __init__(self, hf_model_config):
             self.config.tabular_config = tabular_config.__dict__
 
         tabular_config.text_feat_dim = hf_model_config.hidden_size
-        tabular_config.hidden_dropout_prob = hf_model_config.hidden_dropout_prob
+        tabular_config.dropout = hf_model_config.dropout
         self.tabular_combiner = TabularFeatCombiner(tabular_config)
         self.num_labels = tabular_config.num_labels
         combined_feat_dim = self.tabular_combiner.final_out_dim