testin the trick

modeling_stacked.py

@@ -52,7 +52,9 @@ class ExtendedMultitaskModelForTokenClassification(PreTrainedModel):
 
     def load_state_dict(self, state_dict, strict=True):
         # Ignore loading since there are no parameters
-        print("Ignoring state_dict since model has no parameters.")
+        pass
+
+    # print("Ignoring state_dict since model has no parameters.")
 
     def get_floret_model(self):
         return self.model_floret
@@ -73,7 +75,7 @@ class ExtendedMultitaskModelForTokenClassification(PreTrainedModel):
 
     @classmethod
     def from_pretrained(cls, *args, **kwargs):
-        print("Ignoring weights and using custom initialization.")
+        # print("Ignoring weights and using custom initialization.")
         # Manually create the config
         config = ImpressoConfig(**kwargs)
         # Pass the manually created config to the class