AI开发平台MODELARTS-SD1.5文生图Finetune高性能训练适配NPU指导:Step4 下载模型和数据集

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

Step4 下载模型和数据集

数据集下载地址:https://huggingface.co/datasets/lambdalabs/pokemon-blip-captions

启动脚本前两个声明为本次训练的模型和数据集,第一次执行程序时若本地没有模型和数据集,会自动下载。但由于lambdalabs/pokemon-blip-captions数据集下载现在需要登录HuggingFace账号,请先下载到本地,再挂载到对应目录。
export MODEL_NAME="runwayml/stable-diffusion-v1-5"
export DATASET_NAME="lambdalabs/pokemon-blip-captions"
support.huaweicloud.com/bestpractice-modelarts/modelarts_10_1502.html