Update modeling_colgranitevision.py

modeling_colgranitevision.py

@@ -11,13 +11,13 @@ from .colgranitevision_config import ColGraniteVisionConfig
 
 
 class LlavaNextWithCustomPacking(LlavaNextForConditionalGeneration):
+    
     def pack_image_features(
             self,
             image_features,
             image_sizes,
             vision_feature_select_strategy,
-            image_newline=None,
-            base_image_feature_location="last",
+            image_newline=None
     ):
         """
         Reshape, unpad and then pack each image_feature into a single image_features tensor containing all visual vectors.
@@ -37,6 +37,7 @@ class LlavaNextWithCustomPacking(LlavaNextForConditionalGeneration):
                 token length of each image in image_features
         """
 
+        base_image_feature_location = self.config.base_image_feature_location
         new_image_features = []
         feature_lens = []
         for image_idx, image_feature in enumerate(image_features):