Cyril666 commited on
Commit
50e9f5a
·
verified ·
1 Parent(s): f1d418e

Update image_processing_videollama3.py

Browse files
Files changed (1) hide show
  1. image_processing_videollama3.py +4 -1
image_processing_videollama3.py CHANGED
@@ -39,7 +39,6 @@ from transformers.image_utils import (
39
  ChannelDimension,
40
  ImageInput,
41
  PILImageResampling,
42
- VideoInput,
43
  get_image_size,
44
  infer_channel_dimension_format,
45
  is_scaled_image,
@@ -47,6 +46,10 @@ from transformers.image_utils import (
47
  make_list_of_images,
48
  to_numpy_array,
49
  )
 
 
 
 
50
  from transformers.utils import TensorType, is_vision_available, logging
51
 
52
 
 
39
  ChannelDimension,
40
  ImageInput,
41
  PILImageResampling,
 
42
  get_image_size,
43
  infer_channel_dimension_format,
44
  is_scaled_image,
 
46
  make_list_of_images,
47
  to_numpy_array,
48
  )
49
+ try:
50
+ from transformers.image_utils import VideoInput
51
+ except:
52
+ from transformers.video_utils import VideoInput
53
  from transformers.utils import TensorType, is_vision_available, logging
54
 
55