Ddpm Mnist

🚀 無條件MNIST DDPM

這是一個基於MNIST數據集訓練的輕量級無條件圖像生成模型，可快速生成手寫數字圖像。

🚀 快速開始

本模型是在MNIST數據集上訓練的輕量級UNet2D模型。該模型是無條件的，這意味著你無法指定要生成的數字。此模型在Google Colab的L4 GPU實例上約40分鐘即可完成訓練。你可以在訓練指標標籤中查看訓練日誌。

有條件版本的模型可在 1aurent/ddpm-mnist-conditional 找到，不過它存在一些問題。

💻 使用示例

基礎用法

from diffusers import DDPMPipeline

pipeline = DDPMPipeline.from_pretrained('1aurent/ddpm-mnist')
image = pipeline().images[0]
image

📄 許可證

本項目採用MIT許可證。