import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("Mau124/diffusion-model-cartoon10k-96", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]Model Card for Simple Difussion model using Hugging Face diffusers library
This model is a diffusion model for unconditional image generation of cartoon faces using the cartoon10k dataset. The code for this model can be found on github.
The pipeline for this model was created following Diffusion Models Class 🧨
Usage
from diffusers import DDPMPipeline
pipeline = DDPMPipeline.from_pretrained('Mau124/diffusion-model-cartoon10k-96')
image = pipeline().images[0]
image
Results of generating randomly 16 96x96 images:
- Downloads last month
- -
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support