/项目地址
https:///playgroundai/playground-v2.5-1024px-aesthetic
/
from diffusers import DiffusionPipelineimport torch
pipe = DiffusionPipeline.from_pretrained(“playgroundai/playground-v2.5-1024px-aesthetic”,torch_dtype=torch.float16,variant=”fp16″,).to(“cuda”)
# # Optional: Use DPM++ 2M Karras scheduler for crisper fine details# from diffusers import EDMDPMSolverMultistepScheduler# pipe.scheduler = EDMDPMSolverMultistepScheduler()
prompt = “Astronaut in a jungle, cold color palette, muted colors, detailed, 8k”image = pipe(prompt=prompt, num_inference_steps=50, guidance_scale=3).images[0]
项目链接