import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("sayakpaul/test-kerascv_sd_diffusers_pipeline", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]KerasCV Stable Diffusion in Diffusers 🧨🤗
The pipeline contained in this repository was created using this Space. The purpose is to convert the KerasCV Stable Diffusion weights in a way that is compatible with Diffusers. This allows users to fine-tune using KerasCV and use the fine-tuned weights in Diffusers taking advantage of its nifty features (like schedulers, fast attention, etc.).
- Downloads last month
- 3
Model tree for sayakpaul/test-kerascv_sd_diffusers_pipeline
Base model
CompVis/stable-diffusion-v1-4