README.md

from transformers import pipeline

加载模型

generator = pipeline("text-to-image", model="runwayml/stable-diffusion-v1-5")

定义提示词

prompt = ""

生成图像

image = generator(prompt, num_inference_steps=50, guidance_scale=7.5) image[0].save("my_image.png")