Diffusion Models Class ? 课程单元2的示例微调模型

该模型是从 https://huggingface.co/google/ddpm-bedroom-256 中初始化的扩散模型，在 https://huggingface.co/datasets/huggan/wikiart 上进行5000步的训练。脚本： https://github.com/huggingface/diffusion-models-class/blob/main/unit2/finetune_model.py 训练日志（包含示例图像）： https://wandb.ai/johnowhitaker/dm_finetune/runs/2upaa341

用法

from diffusers import DDPMPipeline

pipeline = DDPMPipeline.from_pretrained('johnowhitaker/sd-class-wikiart-from-bedrooms')
image = pipeline().images[0]
image