Fix import error and add source_files to config
Browse files- foundation_bert.py +2 -2
foundation_bert.py
CHANGED
|
@@ -147,7 +147,7 @@ class FoundationBert(BertModel):
|
|
| 147 |
# 我们显式地请求下载 train_config.yaml
|
| 148 |
try:
|
| 149 |
# 这一步会检查缓存,如果没有则从云端下载并返回本地绝对路径
|
| 150 |
-
|
| 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: {
|
| 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 |
|