testin the trick

modeling_stacked.py

@@ -53,8 +53,8 @@ class ExtendedMultitaskModelForTokenClassification(PreTrainedModel):
 
         # Load floret model
         self.dummy_param = nn.Parameter(torch.zeros(1))
-        self.safe_floret = SafeFloretWrapper(self.config.filename)
-        print(self.config.config)
+        self.safe_floret = SafeFloretWrapper(self.config.config.filename)
+        # print(self.config.config)
         print(type(self.config))
         print(self.config.config.filename)
         # self.model_floret = SafeFloretWrapper(model_floret)