许可协议: MIT 标签:

• pytorch
• diffusers
• 无条件图像生成
• 扩散模型课程

扩散模型课程 🧨 第二单元微调示例模型

该扩散模型基于 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