Text-to-Image
Diffusers
Safetensors
English
FlaxStableDiffusionPipeline
TPU
JAX
Flax
stable-diffusion
import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("camenduru/plushies", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]flax 🧨 checkpoints are here. ckpt and 🧨 pytorch checkpoints here 🎉🎊 https://huggingface.co/camenduru/plushies-pt
Trained with google cloud TPUs.
Runtime: 3h 26m 44s
Steps: 18000
Precision: bf16
Learning Rate: 1e-6
- Downloads last month
- 6