Disable benchmarking mode

argmaxinc · Aug 14, 2024 · b332a04 · b332a04
1 parent 8384f40
commit b332a04
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/python/src/diffusionkit/mlx/scripts/generate_images.py b/python/src/diffusionkit/mlx/scripts/generate_images.py
@@ -112,9 +112,10 @@ def cli():
         args.cfg = 0.0
 
     if args.benchmark_mode:
-        if args.low_memory_mode:
-            logger.warning("Benchmark mode is enabled, disabling low memory mode.")
-        args.low_memory_mode = False
+        raise NotImplementedError
+        # if args.low_memory_mode:
+        #     logger.warning("Benchmark mode is enabled, disabling low memory mode.")
+        # args.low_memory_mode = False
 
     if args.denoise < 0.0 or args.denoise > 1.0:
         raise ValueError("Denoising factor must be between 0.0 and 1.0")