Image Generation
Collection
All of my image generator paraphernalia in one place! • 14 items • Updated • 3
import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("nroggendorff/zelda-diffusion", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]SDZelda is a latent text-to-image diffusion model capable of generating images of Zelda from The Legend of Zelda. 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 StableDiffusionPipeline
import torch
pipeline = StableDiffusionPipeline.from_pretrained("nroggendorff/zelda-diffusion", torch_dtype=torch.float16, use_safetensors=True).to("cuda")
image = pipeline(prompt="a drawing of a woman in a blue dress and gold crown").images[0]
image.save("zelda.png")
train_batch_size: 1gradient_accumulation_steps: 4learning_rate: 1e-2lr_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.