Text-to-Image
Diffusers
Safetensors
StableDiffusionPipeline
stable-diffusion
stable-diffusion-diffusers
import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("digiplay/kotosmix_diffusers", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]model info : https://civitai.com/models/5245/kotosmix
you can apply VAE to get better color,
example codes for diffusers:
#VAE
from diffusers.models import AutoencoderKL
vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse")
modelid="digiplay/kotosmix_diffusers"
pipe = DiffusionPipeline.from_pretrained(modelid, vae=vae)
PS: Recommended Euler scheduler type.
- Downloads last month
- 17