metadata

license: mit
tags:
  - pytorch
  - diffusers
  - unconditional-image-generation
  - diffusion-models-class

Example Fine-Tuned Model for Unit 2 of the Diffusion Models Class 🧨

This model is a diffusion model for unconditional image generation of Ukiyo-e images ✍ 🎨. The model was train using fine-tuning with the google/ddpm-celebahq-256 pretrain-model and the dataset: https://huggingface.co/datasets/huggan/ukiyoe2photo

Usage

from diffusers import DDPMPipeline

pipeline = DDPMPipeline.from_pretrained('alkzar90/sd-class-ukiyo-e-256')
image = pipeline().images[0]
image