Bump version and add ability to guide segmentation with mask.

roboflow · Dec 1, 2023 · 392f424 · 392f424
1 parent faaa22e
commit 392f424
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/maestro/markers/sam.py b/maestro/markers/sam.py
@@ -62,6 +62,7 @@ def generate(
                     input_points=[[input_points]],
                     return_tensors="pt"
                 ).to(self.device)
+                del inputs["pixel_values"]
                 outputs = self.model(image_embeddings=image_embeddings, **inputs)
                 mask = self.processor.image_processor.post_process_masks(
                     masks=outputs.pred_masks.cpu().detach(),

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "maestro"
-version = "0.1.0"
+version = "0.1.1rc1"
 description = "Visual Prompting for Large Multimodal Models (LMMs)"
 authors = ["Piotr Skalski <piotr.skalski92@gmail.com>"]
 maintainers = ["Piotr Skalski <piotr.skalski92@gmail.com>"]