Skip to content

Commit c61a077

Browse files
authored
fix bug for kunlunxin run sd demo for uing fp16 (PaddlePaddle#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
1 parent 409c5dc commit c61a077

File tree

1 file changed

+1
-1
lines changed
  • examples/multimodal/stable_diffusion

1 file changed

+1
-1
lines changed

examples/multimodal/stable_diffusion/infer.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -324,7 +324,7 @@ def get_scheduler(args):
324324
start = time.time()
325325
unet_runtime = create_kunlunxin_runtime(
326326
args.model_dir, args.unet_model_prefix,
327-
args.ues_fp16,
327+
use_fp16=args.ues_fp16,
328328
device_id=args.device_id)
329329
print(f"Spend {time.time() - start : .2f} s to load unet model.")
330330
pipe = StableDiffusionFastDeployPipeline(

0 commit comments

Comments
 (0)