NicFromLM
/

MagiModel

Object Detection

Model card Files Files and versions Community

NicFromLM commited on Jan 4

Commit

24c9704

·

verified ·

1 Parent(s): b3a8991

Update modelling_magiv2.py

Files changed (1) hide show

modelling_magiv2.py +1 -1

modelling_magiv2.py CHANGED Viewed

@@ -54,7 +54,7 @@ class Magiv2Model(PreTrainedModel):
             self.class_labels_classifier = nn.Linear(
                 config.detection_model_config.d_model, config.detection_model_config.num_labels
             )
-                        self.is_this_text_a_dialogue = nn.Linear(
                 config.detection_model_config.d_model, 1
             )
             self.matcher = ConditionalDetrModel(ConditionalDetrConfig(

             self.class_labels_classifier = nn.Linear(
                 config.detection_model_config.d_model, config.detection_model_config.num_labels
             )
+            self.is_this_text_a_dialogue = nn.Linear(
                 config.detection_model_config.d_model, 1
             )
             self.matcher = ConditionalDetrModel(ConditionalDetrConfig(