feat: update processing_valley.py
Browse files- processing_valley.py +1 -2
processing_valley.py
CHANGED
|
@@ -251,7 +251,6 @@ class ValleyProcessor(ProcessorMixin):
|
|
| 251 |
def __call__(self, messages, inference=True, **kwargs) -> BatchFeature:
|
| 252 |
max_pixels=kwargs.get("max_pixels", self.max_pixels)
|
| 253 |
min_pixels=kwargs.get("min_pixels", self.min_pixels)
|
| 254 |
-
print(f"max_pixels: {max_pixels}, min_pixels: {min_pixels}")
|
| 255 |
if max_pixels is not None:
|
| 256 |
self.qwen2vl_image_processor.max_pixels = max_pixels
|
| 257 |
if min_pixels is not None:
|
|
@@ -263,7 +262,7 @@ class ValleyProcessor(ProcessorMixin):
|
|
| 263 |
elif type(messages["images"]) == str:
|
| 264 |
images = [messages["images"]]
|
| 265 |
else:
|
| 266 |
-
images = messages["images"]
|
| 267 |
|
| 268 |
# Deal with conversations
|
| 269 |
conversations = messages["conversations"]
|
|
|
|
| 251 |
def __call__(self, messages, inference=True, **kwargs) -> BatchFeature:
|
| 252 |
max_pixels=kwargs.get("max_pixels", self.max_pixels)
|
| 253 |
min_pixels=kwargs.get("min_pixels", self.min_pixels)
|
|
|
|
| 254 |
if max_pixels is not None:
|
| 255 |
self.qwen2vl_image_processor.max_pixels = max_pixels
|
| 256 |
if min_pixels is not None:
|
|
|
|
| 262 |
elif type(messages["images"]) == str:
|
| 263 |
images = [messages["images"]]
|
| 264 |
else:
|
| 265 |
+
images = messages["images"]
|
| 266 |
|
| 267 |
# Deal with conversations
|
| 268 |
conversations = messages["conversations"]
|