Cyril666 commited on
Commit
88cc348
·
verified ·
1 Parent(s): 3fe100e

Update processing_videollama3.py

Browse files
Files changed (1) hide show
  1. processing_videollama3.py +1 -1
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