Sd Class Wikiart From Bedrooms

🚀 擴散模型課程單元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

📄 許可證

本項目採用MIT許可證。