Downloads · 30 days
639
7% of all-time downloads
nroggendorff/cats
cats is a unconditional image generation model from nroggendorff. Use it for the unconditional image generation task on the model card, and read the license before you ship it in a product. It is set up for diffusers. The card lists the license as mit.
DDPMCats is a latent noise-to-image diffusion model capable of generating images of cats. For more information about how Stable Diffusion functions, please have a look at 🤗's Stable Diffusion blog.
Downloads · 30 days
639
7% of all-time downloads
All-time downloads
9.1K
Public
Parameters
114M
13.1 GB on disk
Likes
1
Public
Click a slice to open those files.
.safetensors3.2 GB · 87%
From the Hugging Face model README
DDPMCats is a latent noise-to-image diffusion model capable of generating images of cats. For more information about how Stable Diffusion functions, please have a look at 🤗's Stable Diffusion blog.
You can use this with the 🧨Diffusers library from Hugging Face.

from diffusers import DiffusionPipeline
pipeline = DiffusionPipeline.from_pretrained("nroggendorff/cats")
pipe = pipeline.to("cuda")
image = pipe().images[0]
image.save("cat.png")
train_batch_size: 16eval_batch_size: 16num_epochs: 50gradient_accumulation_steps: 1learning_rate: 1e-4lr_warmup_steps: 500mixed_precision: "fp16"eval_metric: "mean_squared_error"This model card was written by Noa Roggendorff and is based on the Stable Diffusion v1-5 Model Card.