diff --git a/src/transformers/models/vivit/image_processing_vivit.py b/src/transformers/models/vivit/image_processing_vivit.py index 0790c5d82bb5..af02400e71e6 100644 --- a/src/transformers/models/vivit/image_processing_vivit.py +++ b/src/transformers/models/vivit/image_processing_vivit.py @@ -83,7 +83,7 @@ class VivitImageProcessor(BaseImageProcessor): do_rescale (`bool`, *optional*, defaults to `True`): Whether to rescale the image by the specified scale `rescale_factor`. Can be overridden by the `do_rescale` parameter in the `preprocess` method. - rescale_factor (`int` or `float`, *optional*, defaults to `1/255`): + rescale_factor (`int` or `float`, *optional*, defaults to 1/127.5): Defines the scale factor to use if rescaling the image. Can be overridden by the `rescale_factor` parameter in the `preprocess` method. offset (`bool`, *optional*, defaults to `True`): @@ -110,7 +110,7 @@ def __init__( do_center_crop: bool = True, crop_size: Dict[str, int] = None, do_rescale: bool = True, - rescale_factor: Union[int, float] = 1 / 255, + rescale_factor: Union[int, float] = 1 / 127.5, offset: bool = True, do_normalize: bool = True, image_mean: Optional[Union[float, List[float]]] = None,