Spaces:
Paused
Paused
Commit ·
249e4fa
1
Parent(s): edac724
update
Browse files
examples/tutorials/lora_transformers/step_2_train_model.py
CHANGED
|
@@ -170,7 +170,7 @@ def main():
|
|
| 170 |
|
| 171 |
# ==================== 4.保存训练结果 ====================================
|
| 172 |
# 只保存lora适配器参数
|
| 173 |
-
trained_models_dir = project_path / "trained_models" / "Qwen3-8B-sft-lora-adapter-
|
| 174 |
trained_models_dir.mkdir(parents=True, exist_ok=True)
|
| 175 |
model.save_pretrained(trained_models_dir.as_posix())
|
| 176 |
tokenizer.save_pretrained(trained_models_dir.as_posix())
|
|
|
|
| 170 |
|
| 171 |
# ==================== 4.保存训练结果 ====================================
|
| 172 |
# 只保存lora适配器参数
|
| 173 |
+
trained_models_dir = project_path / "trained_models" / "Qwen3-8B-sft-lora-adapter-transformers"
|
| 174 |
trained_models_dir.mkdir(parents=True, exist_ok=True)
|
| 175 |
model.save_pretrained(trained_models_dir.as_posix())
|
| 176 |
tokenizer.save_pretrained(trained_models_dir.as_posix())
|