单机多卡部署Stable Diffusion： 注意 diffusers版本

# 替换为你模型文件夹的本地路径
    MODEL_PATH = "/data1/fyu/models/stable-diffusion-v1-5"

    # 加载本地模型并自动分配到多张GPU
    pipe = StableDiffusionPipeline.from_pretrained(
        MODEL_PATH,
        torch_dtype=torch.float16,
        revision="fp16",
        device_map="balanced" # 自动切分模型到多GPU
    )

    # 设置提示词
    prompt = "A futuristic city with flying cars, concept art, sunset, 4k"

    # 生成图像
    image = pipe(prompt, guidance_scale=7.5, num_inference_steps=30).images[0]

    # 保存图像
    image.save("output.png")
    print("成功生成图像：output.png")

原环境的 diffusers版本为 0.24.0，报错信息如下

 diffusers升级为 0.34.0后，成功运行