File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed
src/transformers/models/vilt Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change 14
14
# limitations under the License.
15
15
"""Fast Image processor class for Vilt."""
16
16
17
- from typing import Optional
17
+ from typing import List , Optional , Union
18
18
19
19
from ...image_processing_utils import BatchFeature
20
20
from ...image_processing_utils_fast import (
@@ -88,9 +88,9 @@ def _preprocess(
88
88
do_rescale : bool ,
89
89
rescale_factor : float ,
90
90
do_normalize : bool ,
91
- image_mean : Optional [float | list [float ]],
92
- image_std : Optional [float | list [float ]],
93
- return_tensors : Optional [str | TensorType ],
91
+ image_mean : Optional [Union [ float , List [float ] ]],
92
+ image_std : Optional [Union [ float , List [float ] ]],
93
+ return_tensors : Optional [Union [ str , TensorType ] ],
94
94
** kwargs ,
95
95
) -> BatchFeature :
96
96
"""
You can’t perform that action at this time.
0 commit comments