AI开发平台MODELARTS-SD1.5基于DevServer适配PyTorch NPU Finetune训练指导（6.3.904）:Step3 获取SD1.5插件代码包并安装依赖

时间：2025-02-20 08:32:13

AI开发平台MODELARTS

Step3 获取SD1.5插件代码包并安装依赖

将下载的SD1.5插件代码包ascendcloud-aigc-xxx-xxx.tar.gz文件，上传到容器的/home/ma-user/目录下，解压并安装相关依赖。插件代码包获取路径参见表2。

mkdir -p /home/ma-user/stable_diffusers_1.5    #创建stable_diffusers_1.5目录
cd /home/ma-user/stable_diffusers_1.5          #进入stable_diffusers_1.5目录

tar -zxvf ascendcloud-aigc-*.tar.gz
tar -zxvf ascendcloud-aigc-poc-stable_diffusers_1.5.tar.gz
rm -rf ascendcloud-aigc-xxx-xxx

pip install -r requirements.txt    #安装依赖

启动前配置。有两种方式修改配置文件：
- 方式一：可以参考解压出来的default_config.yaml或者deepspeed_default_config.yaml文件，再通过在启动脚本命令中增加--config_file=xxx.yaml参数来指定其为配置文件。
- 方式二：通过命令accelerate config进行配置，如下图所示。
  图1 通过命令accelerate config进行配置
（可选）文件替换。
因增加nfa和使用npu_geglu算子（用于训练和推理加速），将diffusers源码包中的attention.py和attention_processor.py替换成代码包中对应的文件。

图2 文件替换

可以使用find命令来查找diffusers源码包位置。
```
find / -name attention.py
find / -name attention_processor.py
```
图3 查找diffusers源码包位置

找到具体位置后可以cp替换，替换前可对diffusers原始文件做备份，如果没有备份则可以通过删除diffusers包重新安装的方式获取原始文件。
执行bash stable_diffusers_train.sh。
```
bash stable_diffusers_train.sh
```