From c61a07712e354e6f0e24329fecbc03433ec3d7c5 Mon Sep 17 00:00:00 2001
From: wangguoya <39376046+wgy0804@users.noreply.github.com>
Date: Mon, 27 Mar 2023 14:04:21 +0800
Subject: [PATCH] fix bug for kunlunxin run sd demo for uing fp16 (#1680)

* modify sd infer.py for using paddle_kunlunxin_fp16

* Update infer.py

* [fix bug] fix bug sd in demo infer.py for kunlunxin using fp16
---
 examples/multimodal/stable_diffusion/infer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/multimodal/stable_diffusion/infer.py b/examples/multimodal/stable_diffusion/infer.py
index 001f67864..d2a8fb0cd 100755
--- a/examples/multimodal/stable_diffusion/infer.py
+++ b/examples/multimodal/stable_diffusion/infer.py
@@ -324,7 +324,7 @@ if __name__ == "__main__":
         start = time.time()
         unet_runtime = create_kunlunxin_runtime(
             args.model_dir, args.unet_model_prefix, 
-            args.ues_fp16,
+            use_fp16=args.ues_fp16,
             device_id=args.device_id)
         print(f"Spend {time.time() - start : .2f} s to load unet model.")
     pipe = StableDiffusionFastDeployPipeline(