change grid size to 8 (align with SD)

codeinvain · Jan 16, 2024 · dce1450 · dce1450
1 parent f90fb34
commit dce1450
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/node.py b/node.py
@@ -403,9 +403,12 @@ def INPUT_TYPES(cls):
     CATEGORY = "segment_anything"
 
     def load_model(self, model_name):
+        model = VitMatteForImageMatting.from_pretrained(model_name)
+        processor = VitMatteImageProcessor.from_pretrained(model_name)
+        processor.size_divisibility = 8 # align with SD grid size
         vitmatte = VITMatteModel(
-            VitMatteForImageMatting.from_pretrained(model_name),
-            VitMatteImageProcessor.from_pretrained(model_name)
+            model,
+            processor,
         )
         return (vitmatte,)