AI开发平台MODELARTS-SD3基于Lite Server适配PyTorch NPU的训练指导(6.3.912):步骤三:开始训练

时间:2025-02-25 19:55:15

步骤三:开始训练

  1. 进入容器中/home/ma-user/aigc_train/torch_npu/sd3路径下
    cd /home/ma-user/aigc_train/torch_npu/sd3
  2. 安装依赖
    sh prepare.sh

    如果这一步安装依赖失败,是部分依赖之间有冲突,手动在终端依次执行如下命令解决。

    pip install wandb
    pip install urllib3==1.26.7
    cp run.sh diffusers/examples/dreambooth/
  3. 开始训练。
    1. 进入diffusers/examples/dreambooth
      cd diffusers/examples/dreambooth
    2. 修改run.sh文件的前两行为模型权重和数据集路径。bs,step,resolution用给定参数。

    3. 运行run.sh
      sh run.sh
  4. 训练过程中,终端弹出wandb选项,如下图所示,输入3即可。

    训练成功如下图所示。

support.huaweicloud.com/bestpractice-modelarts/modelarts_sd3_train_912.html