Fix (example/sdxl): Send device to fid computation backend

Xilinx · Sep 5, 2024 · ac95519 · ac95519
1 parent fb20e00
commit ac95519
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/brevitas_examples/stable_diffusion/main.py b/src/brevitas_examples/stable_diffusion/main.py
@@ -520,7 +520,7 @@ def input_zp_stats_type():
         if args.use_mlperf_inference:
             print(f"Computing accuracy with MLPerf pipeline")
             compute_mlperf_fid(
-                args.model, args.path_to_coco, pipe, args.prompt, output_dir, not args.vae_fp16_fix)
+                args.model, args.path_to_coco, pipe, args.prompt, output_dir, args.device, not args.vae_fp16_fix)
         else:
             print(f"Computing accuracy on default prompt")
             testing_prompts = TESTING_PROMPTS[:args.prompt]