We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 409c5dc commit c61a077Copy full SHA for c61a077
examples/multimodal/stable_diffusion/infer.py
@@ -324,7 +324,7 @@ def get_scheduler(args):
324
start = time.time()
325
unet_runtime = create_kunlunxin_runtime(
326
args.model_dir, args.unet_model_prefix,
327
- args.ues_fp16,
+ use_fp16=args.ues_fp16,
328
device_id=args.device_id)
329
print(f"Spend {time.time() - start : .2f} s to load unet model.")
330
pipe = StableDiffusionFastDeployPipeline(
0 commit comments