handle non-RGB images

vikhyat · Dec 5, 2024 · 8a622f0 · 8a622f0
1 parent f3304d1
commit 8a622f0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/clients/python/moondream/preprocess.py b/clients/python/moondream/preprocess.py
@@ -78,6 +78,8 @@ def create_patches(
     resolution. Returns the patches as a numpy array, and the selected patching
     template as a tuple of (rows, cols).
     """
+    image = image.convert("RGB")
+
     # Start off with the global patch.
     patches = [im_resize(image, (image_patch_size, image_patch_size))]
 

diff --git a/clients/python/pyproject.toml b/clients/python/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "moondream"
-version = "0.0.4"
+version = "0.0.5"
 description = "Python client library for moondream"
 authors = ["vik <[email protected]>"]
 readme = "README.md"