AI开发平台MODELARTS-SDXL基于DevServer适配PyTorch NPU的Finetune训练指导（6.3.905）:Step2 下载代码包、依赖模型包和数据集

时间：2024-12-09 20:36:09

AI开发平台MODELARTS

Step2 下载代码包、依赖模型包和数据集

下载stable-diffusion-xl-base-1.0模型包并上传到宿主机上，官网下载地址：https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/tree/main
下载vae-fp16-fix模型包并上传到宿主机上，官网下载地址：https://huggingface.co/madebyollin/sdxl-vae-fp16-fix/tree/main
下载开源数据集并上传到宿主机上，官网下载地址：https://huggingface.co/datasets/lambdalabs/pokemon-blip-captions/tree/main。用户也可以使用自己的数据集。

下载SDXL插件代码包AscendCloud-3rdAIGC-6.3.905-xxx.zip文件，获取路径参见获取软件和镜像。本案例使用的是AscendCloud-3rdAIGC-6.3.905-xxx.zip文件中的ascendcloud-aigc-poc-sdxl-finetune.tar.gz代码包。解压后上传到宿主机上。

依赖的插件代码包、模型包和数据集存放在宿主机上的本地目录结构如下，供参考。

[root@devserver-ei-cto-office-ae06cae7-tmp1216 docker_build]# ll
total 192
-rw-------  1 root root 108286 May  6 16:56 attention_processor.py
-rw-------  1 root root    430 May  8 09:31 config.yaml
drwx------  3 root root   4096 May  7 10:50 datasets
-rw-------  1 root root   1356 May  8 16:30 diffusers_finetune_train.sh
-rw-------  1 root root   1468 May  8 16:49 Dockerfile    #需要用户参考Step3构建镜像步骤写Dockerfile文件
drwx------ 10 root root   4096 Apr 30 15:18 stable-diffusion-xl-base-1.0
-rw-------  1 root root  58048 May  8 17:48 train_text_to_image_sdxl-0212.py
drwx------  2 root root   4096 Apr 30 15:17 vae-fp16-fix

上一篇：AI开发平台MODELARTS-SDXL基于DevServer适配PyTorch NPU的Finetune训练指导（6.3.905）:获取软件和镜像

下一篇：AI开发平台MODELARTS-SDXL基于DevServer适配PyTorch NPU的Finetune训练指导（6.3.905）:获取软件和镜像