README.md

metadata

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

Model Card for Unit 1 of the Diffusion Models Class 🧨

This model is a diffusion model for unconditional image generation of cute 🦋.

Usage

from diffusers import DDPMPipeline
import torch

pipeline = DDPMPipeline.from_pretrained('Chen011/sd-class-butterflies-32').to("cuda" if torch.cuda.is_available() else "cpu")

# Generate random noise
random_noise = torch.randn((1, 3, 64, 64))  # Adjust the shape as per your model's requirements

# Pass through the pipeline
image = pipeline(num_inference_steps=50, generator=torch.manual_seed(42), latents=random_noise).images[0]
image.show()