Merge pull request #31 from argmaxinc/check_res_div

Check image dim divisibility by 16

python/src/diffusionkit/mlx/__init__.py

@@ -298,6 +298,13 @@ def generate_image(
         image_path: Optional[str] = None,
         denoise: float = 1.0,
     ):
+        # Check latent size is divisible by 2
+        assert (
+            latent_size[0] % 2 == 0
+        ), f"Height must be divisible by 16 ({latent_size[0]*8}/16={latent_size[0]/2})"
+        assert (
+            latent_size[1] % 2 == 0
+        ), f"Width must be divisible by 16 ({latent_size[1]*8}/16={latent_size[1]/2})"
         self.check_and_load_models()
         # Start timing
         start_time = time.time()

python/src/diffusionkit/mlx/scripts/generate_images.py

@@ -142,6 +142,8 @@ def cli():
 
     height = args.height or HEIGHT[args.model_version]
     width = args.width or WIDTH[args.model_version]
+    assert height % 16 == 0, f"Height must be divisible by 16 ({height}/16={height/16})"
+    assert width % 16 == 0, f"Width must be divisible by 16 ({width}/16={width/16})"
     logger.info(f"Output image resolution will be {height}x{width}")
 
     if args.benchmark_mode:

setup.py

@@ -3,7 +3,7 @@
 from setuptools import find_packages, setup
 from setuptools.command.install import install
 
-VERSION = "0.3.4"
+VERSION = "0.3.5"
 
 
 class VersionInstallCommand(install):