jaimemedina/ddpm-cat-64-optimized

DDPM Cat Generator (64x64)

This model is a Denoising Diffusion Probabilistic Model (DDPM) trained on the AFHQ (Animal Faces HQ) - Cat dataset. It generates 64x64 pixel images of cat faces.

It was trained using the 🤗 Diffusers library and PyTorch.

Usage

You can use this model directly with the diffusers library to generate new images.

from diffusers import DDPMPipeline
import torch

# Load the pipeline
pipeline = DDPMPipeline.from_pretrained("jaimemedina/ddpm-cat-64-optimized")

# Move to GPU if available
pipeline.to("cuda" if torch.cuda.is_available() else "cpu")

# Generate an image
image = pipeline(num_inference_steps=1000).images[0]
image.save("cute_cat.png")