AI开发平台MODELARTS-SDXL Diffusers框架基于DevServer适配PyTorch NPU推理指导(6.3.902):Step2 安装依赖和模型包

时间:2024-11-12 16:42:21

Step2 安装依赖和模型包

  1. 安装Diffusers相关依赖。
    pip install -i https://pypi.tuna.tsinghua.edu.cn/simple diffusers bottle invisible_watermark transformers accelerate safetensors
  2. 获取SDXL模型包并解压到/home/ma-user目录下。提供2种模型包下载方式。
    • 模型包直接下载(如果不能访问HuggingFace官网,推荐此方式)
      下载到容器/home/ma-user目录下后,解压。
      cd  /home/ma-user/ 
      wget https://llm-mindspore.obs.cn-southwest-2.myhuaweicloud.com/ascend-poc/stable-diffusion-xl-model.tar.gz
      tar -zxvf stable-diffusion-xl-model.tar.gz 
      rm -rf stable-diffusion-xl-model.tar.gz
    • 也可以从HuggingFace官网下载到本地后,通过docker cp命令复制到容器中/home/ma-user目录下,如下图所示。

      在线下载地址:

      https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/tree/main

      https://huggingface.co/stabilityai/stable-diffusion-xl-refiner-1.0/tree/main

      由于本实例采用的都是FP16的模型,相应模型建议都只下载FP16的,节约下载和传送时间。

      图1 下载SDXL模型包并解压
  3. 获取controlnet模型包并解压到/home/ma-user目录下。提供2种模型包下载方式。
    • 模型包直接下载(如果不能访问HuggingFace官网,推荐此方式)
      下载到容器/home/ma-user目录下后,解压。
      cd  /home/ma-user/ 
      wget https://llm-mindspore.obs.cn-southwest-2.myhuaweicloud.com/ascend-poc/controlnet_canny.zip
      unzip controlnet_canny.zip
    • 也可以从HuggingFace官网下载到本地后,通过docker cp命令复制到容器中/home/ma-user目录下。

      在线下载地址:https://huggingface.co/diffusers/controlnet-canny-sdxl-1.0/tree/main

      图2 下载controlnet模型包并解压
  4. 安装插件代码包。
    1. 将获取到的插件代码包ascendcloud-aigc-6.3.902-*.tar.gz文件上传到容器的/home/ma-user/temp目录下。获取路径:Support网站
    2. 解压插件代码包ascendcloud-aigc-6.3.902-*到/home/ma-user/temp目录下。
      cd /home/ma-user/temp
      tar -zxvf ascendcloud-aigc-6.3.902-20240205145924.tar.gz  #解压
    3. 将获取到的ascendcloud-aigc-extensions-diffusers.tar.gz包复制到/home/ma-user下后解压。
      docker cp ascendcloud-aigc-extensions-diffusers.tar.gz sdxl-diffusers:/home/ma-user/
      tar -zxvf ascendcloud-aigc-extensions-diffusers.tar.gz
support.huaweicloud.com/bestpractice-modelarts/modelarts_10_1520.html