当前位置:   article > 正文

SWIFT环境配置及大模型微调实践

SWIFT环境配置及大模型微调实践

SWIFT环境配置及大模型微调实践


SWIFT介绍参考: 这里

SWIFT环境配置

基础配置

conda create -n swift python=3.8

pip install ms-swift[all] -U

# 下载项目
git clone https://github.com/modelscope/swift
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6

增量配置

如果是运行qwen-audio,需要安装ffmpeg

apt-get update
apt-get install ffmpeg
  • 1
  • 2

SWIFT大模型微调实践

也可以参考https://zhuanlan.zhihu.com/p/665300386

微调代码

cd swift/examples/pytorch/llm
nohup bash scripts/qwen_audio_chat/lora/sft.sh > qwen_audio_chat_sft_20240320.log &
  • 1
  • 2

第一次跑,发现使用的是cpu,经过检查,发现pytorch版本对应的cuda版本较高,于是重新安装pytorch

conda install --use-local pytorch-1.13.0-py3.8_cuda11.6_cudnn8.3.2_0.tar.bz2.tar.bz2
  • 1
声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:【wpsshop博客】
推荐阅读