Update processing_videollama3.py
Browse files
processing_videollama3.py
CHANGED
@@ -308,7 +308,7 @@ class Videollama3Qwen2Processor(ProcessorMixin):
|
|
308 |
if isinstance(image_path, list):
|
309 |
images = [load_single_image(f) for f in image_path]
|
310 |
elif isinstance(image_path, str) and os.path.isdir(image_path):
|
311 |
-
images = Image.open(os.path.join(image_path, f)).convert('RGB') for f in sorted(os.listdir(image_path))]
|
312 |
else:
|
313 |
images = [load_single_image(image_path)]
|
314 |
return images
|
|
|
308 |
if isinstance(image_path, list):
|
309 |
images = [load_single_image(f) for f in image_path]
|
310 |
elif isinstance(image_path, str) and os.path.isdir(image_path):
|
311 |
+
images = [Image.open(os.path.join(image_path, f)).convert('RGB') for f in sorted(os.listdir(image_path))]
|
312 |
else:
|
313 |
images = [load_single_image(image_path)]
|
314 |
return images
|