Fix import error and add source_files to config

foundation_bert.py

@@ -141,13 +141,22 @@ class FoundationBert(BertModel):
         use_safetensors: bool = None,
         **kwargs,
     ):
-        """
-            Modification to correctly handle loading extraneous parameters for GBert
-        """
-
-        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}")
+        from huggingface_hub import hf_hub_download
+        
+        # 1. 如果是远程加载，pretrained_model_name_or_path 就是 REPO_ID
+        # 我们显式地请求下载 train_config.yaml
+        try:
+            # 这一步会检查缓存，如果没有则从云端下载并返回本地绝对路径
+            config_file_path = hf_hub_download(
+                repo_id=pretrained_model_name_or_path,
+                filename="train_config.yaml",
+                revision=kwargs.get("revision", "main")
+            )
+        except Exception as e:
+            # 备选方案：如果本地路径已存在（例如 Snigdaa 的用法）
+            model_config = os.path.join(pretrained_model_name_or_path, "train_config.yaml")
+
+        print(f"✅ Successfully located config at: {config_file_path}")
         with open(model_config, 'r') as f:
             config = yaml.load(f, Loader=MyLoader)