昇腾社区首页
中文
注册

大模型迁移

参考迁移前准备,将LLaMA-Factory代码仓目录下的相关文件进行替换,完成迁移操作。

  1. 使用ModelZoo-Pytorch“utils”目录下的train_bash.py文件替换LLaMA-Factory代码仓“src”目录下的train_bash.py文件。
  2. 使用ModelZoo-Pytorch“utils”目录下的misc.py文件替换LLaMA-Factory代码仓“/src/llmtuner/extras”目录下的misc.py文件。
  3. 使用ModelZoo-Pytorch“utils”目录下的modeling_baichuan.py文件替换LLaMA-Factory代码仓“model_weight”目录下的modeling_baichuan.py文件。