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]