From b332a045555cf02166a48b724e4a66d080117619 Mon Sep 17 00:00:00 2001 From: atila Date: Wed, 14 Aug 2024 00:07:13 -0700 Subject: [PATCH] Disable benchmarking mode --- python/src/diffusionkit/mlx/scripts/generate_images.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/python/src/diffusionkit/mlx/scripts/generate_images.py b/python/src/diffusionkit/mlx/scripts/generate_images.py index d48df0f..22228d6 100644 --- 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")