File size: 391 Bytes
871410b 292fc95 871410b 28616f0 292fc95 |
1 2 3 4 5 6 7 8 9 10 |
import torch
from diffusers import AutoPipelineForText2Image
from diffusers import DiffusionPipeline
pipe = DiffusionPipeline.from_pretrained("shuttleai/shuttle-3.1-aesthetic",
torch_dtype=torch.float16,USE_SAFETENSORS=True).to("cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]
|