Views
No views yet
1from diffusers import StableDiffusionOnnxPipeline, PNDMScheduler
2model_path = r'ShadowPower/waifu-diffusion-diffusers-onnx-v1-3'
3
4scheduler = PNDMScheduler(
5 beta_start=0.00085,
6 beta_end=0.012,
7 beta_schedule='scaled_linear',
8 skip_prk_steps=True,
9 tensor_format='np'
10)
11pipe = StableDiffusionOnnxPipeline.from_pretrained(
12 model_path,
13 provider="CPUExecutionProvider",
14 scheduler=scheduler
15)
16
17if __name__ == '__main__':
18 prompt = "1girl, hakurei reimu"
19 image = pipe(prompt).images[0]