Image Generation
Collection
All of my image generator paraphernalia in one place! • 14 items • Updated • 3
How to use nroggendorff/obama with Diffusers:
pip install -U diffusers transformers accelerate
import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("nroggendorff/obama", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]DDPMObama is a latent noise-to-image diffusion model capable of generating images of obama. 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/obama")
pipe = pipeline.to("cuda")
image = pipe().images[0]
image.save("obama.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.