Fix import error and add source_files to config

foundation_bert.py

@@ -144,10 +144,13 @@ class FoundationBert(BertModel):
         """
             Modification to correctly handle loading extraneous parameters for GBert
         """
-        model_config = Path(pretrained_model_name_or_path) / 'train_config.yaml'
-        print(f"🆘 {pretrained_model_name_or_path=}; {model_config=}")
+
+        current_dir = os.path.dirname(os.path.abspath(__file__))
+        model_config = os.path.join(current_dir, 'train_config.yaml')
+        print(f"🆘 DEBUG: Attempting to load config from: {model_config}")
         with open(model_config, 'r') as f:
             config = yaml.load(f, Loader=MyLoader)
+
         kwargs['modalities'] = config['modalities']
         kwargs['dataset_path'] = config['dataset_path']
         kwargs['mask_token'] = config['mask_token']