BugFix VideoInput processing_dreamvl

#2
Files changed (1) hide show
  1. processing_dreamvl.py +7 -1
processing_dreamvl.py CHANGED
@@ -29,7 +29,13 @@ except ImportError:
29
  from typing_extensions import Unpack
30
 
31
  from transformers.feature_extraction_utils import BatchFeature
32
- from transformers.image_utils import ImageInput, VideoInput
 
 
 
 
 
 
33
  from transformers.processing_utils import (
34
  ProcessingKwargs,
35
  ProcessorMixin,
 
29
  from typing_extensions import Unpack
30
 
31
  from transformers.feature_extraction_utils import BatchFeature
32
+ from transformers.image_utils import ImageInput
33
+
34
+ try:
35
+ from transformers.video_utils import VideoInput
36
+ except ImportError as e:
37
+ from trasnformers.image_utils import VideoInput
38
+
39
  from transformers.processing_utils import (
40
  ProcessingKwargs,
41
  ProcessorMixin,