AutoencoderKL 下载

Undefined443 / 2024-11-11 / 原文

from diffusers.models import AutoencoderKL

AutoencoderKL.frome_pretrained("stabilityai/sd/vae-ft-mse")  # 下载 MSE 模型

模型将下载到 ~/.cache/huggingface/hub 目录下。