AI开发平台MODELARTS-SD1.5文生图Finetune高性能训练适配NPU指导:Step3 获取SD1.5插件代码包并安装依赖

时间：2024-06-17 15:07:56

AI开发平台MODELARTS

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

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

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-6.3.*-*.tar.gz
tar -zxvf ascendcloud-aigc-poc-stable_diffusers_1.5.tar.gz
rm -rf ascendcloud-aigc-*

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
```