BugFix VideoInput image_processing_dreamvl

#3
Files changed (1) hide show
  1. image_processing_dreamvl.py +6 -1
image_processing_dreamvl.py CHANGED
@@ -36,7 +36,6 @@ from transformers.image_utils import (
36
  ChannelDimension,
37
  ImageInput,
38
  PILImageResampling,
39
- VideoInput,
40
  get_image_size,
41
  infer_channel_dimension_format,
42
  is_scaled_image,
@@ -46,6 +45,12 @@ from transformers.image_utils import (
46
  valid_images,
47
  validate_preprocess_arguments,
48
  )
 
 
 
 
 
 
49
  from transformers.utils import TensorType, is_vision_available, logging
50
 
51
  logger = logging.get_logger(__name__)
 
36
  ChannelDimension,
37
  ImageInput,
38
  PILImageResampling,
 
39
  get_image_size,
40
  infer_channel_dimension_format,
41
  is_scaled_image,
 
45
  valid_images,
46
  validate_preprocess_arguments,
47
  )
48
+
49
+ try:
50
+ from transformers.video_utils import VideoInput
51
+ except ImportError as e:
52
+ from trasnformers.image_utils import VideoInput
53
+
54
  from transformers.utils import TensorType, is_vision_available, logging
55
 
56
  logger = logging.get_logger(__name__)