Xsmos commited on
Commit
b167f59
·
verified ·
1 Parent(s): fd499f6

Fix import error and add source_files to config

Browse files
Files changed (1) hide show
  1. foundation_bert.py +2 -2
foundation_bert.py CHANGED
@@ -147,7 +147,7 @@ class FoundationBert(BertModel):
147
  # 我们显式地请求下载 train_config.yaml
148
  try:
149
  # 这一步会检查缓存,如果没有则从云端下载并返回本地绝对路径
150
- config_file_path = hf_hub_download(
151
  repo_id=pretrained_model_name_or_path,
152
  filename="train_config.yaml",
153
  revision=kwargs.get("revision", "main")
@@ -156,7 +156,7 @@ class FoundationBert(BertModel):
156
  # 备选方案:如果本地路径已存在(例如 Snigdaa 的用法)
157
  model_config = os.path.join(pretrained_model_name_or_path, "train_config.yaml")
158
 
159
- print(f"✅ Successfully located config at: {config_file_path}")
160
  with open(model_config, 'r') as f:
161
  config = yaml.load(f, Loader=MyLoader)
162
 
 
147
  # 我们显式地请求下载 train_config.yaml
148
  try:
149
  # 这一步会检查缓存,如果没有则从云端下载并返回本地绝对路径
150
+ model_config = hf_hub_download(
151
  repo_id=pretrained_model_name_or_path,
152
  filename="train_config.yaml",
153
  revision=kwargs.get("revision", "main")
 
156
  # 备选方案:如果本地路径已存在(例如 Snigdaa 的用法)
157
  model_config = os.path.join(pretrained_model_name_or_path, "train_config.yaml")
158
 
159
+ print(f"✅ Successfully located config at: {model_config}")
160
  with open(model_config, 'r') as f:
161
  config = yaml.load(f, Loader=MyLoader)
162