File size: 275 Bytes
ad4a1a4 748acdd ad4a1a4 6b96c27 ad4a1a4 1cd266d ad4a1a4 |
1 2 3 4 5 6 7 8 9 10 11 12 |
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")
|