# Copyright 2024 Huawei Technologies Co., Ltd



import torch

import torch_npu

from diffusers import StableDiffusion3Pipeline



pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16)

pipe = pipe.to("npu")



image = pipe(

    "A cat holding a sign that says hello world",

    negative_prompt="",

    num_inference_steps=28,

    guidance_scale=7.0,

).images[0]

image.save("cat.jpg")