AI开发平台MODELARTS-Open-Sora-Plan1.0基于DevServer适配PyTorch NPU训练推理指导（6.3.907）:Step8 下载权重文件

时间：2024-09-24 10:50:23

AI开发平台MODELARTS

Step8 下载权重文件

建议手动下载所需的权重文件，在/home/ma-user/Open-Sora-Plan1.0/目录下进行操作。

创建文件夹存放不同的权重文件。

mkdir weights
mkdir weights_t5
mkdir cache_dir

下载基础模型权重t2v.pt放到cache_dir文件夹下。

mkdir Latte
cd Latte
git clone  -c http.sslVerify=false  https://huggingface.co/maxin-cn/Latte
cd Latte
git reset  --hard  83bdc71f7211963153464859d03d46d707e77865

然后将该目录下的t2v.pt文件复制到/home/ma-user/Open-Sora-Plan1.0/cache_dir目录下。

下载VAE权重vae-ft-mse-840000-ema-pruned.ckpt和配置文件config.json，放在weights文件夹下。
下载链接：https://huggingface.co/stabilityai/sd-vae-ft-ema/tree/main
下载text_encoder权重，放在weights_t5文件夹下。
下载链接：https://huggingface.co/DeepFloyd/t5-v1_1-xxl/tree/main，手动下载如图6所示文件，并放到weights_t5文件夹下

 图6 Huggingface中t5-v1_1-xxl模型目录内容