From af2c36793fd6cde4b595a4b6f800a96dff86556e Mon Sep 17 00:00:00 2001 From: Sheon Han Date: Thu, 1 Jun 2023 07:10:15 -0700 Subject: [PATCH] Fix doc string nits (#23929) --- src/transformers/configuration_utils.py | 2 +- src/transformers/models/vit/image_processing_vit.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/transformers/configuration_utils.py b/src/transformers/configuration_utils.py index 3b7f668b9ea8bc..6e8104c0cd2e66 100755 --- a/src/transformers/configuration_utils.py +++ b/src/transformers/configuration_utils.py @@ -136,7 +136,7 @@ class PretrainedConfig(PushToHubMixin): diversity_penalty (`float`, *optional*, defaults to 0.0): Value to control diversity for group beam search. that will be used by default in the `generate` method of the model. 0 means no diversity penalty. The higher the penalty, the more diverse are the outputs. - temperature (`float`, *optional*, defaults to 1): + temperature (`float`, *optional*, defaults to 1.0): The value used to module the next token probabilities that will be used by default in the `generate` method of the model. Must be strictly positive. top_k (`int`, *optional*, defaults to 50): diff --git a/src/transformers/models/vit/image_processing_vit.py b/src/transformers/models/vit/image_processing_vit.py index 66d68e7f82d75a..71f50d8b9cab71 100644 --- a/src/transformers/models/vit/image_processing_vit.py +++ b/src/transformers/models/vit/image_processing_vit.py @@ -56,7 +56,7 @@ class ViTImageProcessor(BaseImageProcessor): rescale_factor (`int` or `float`, *optional*, defaults to `1/255`): Scale factor to use if rescaling the image. Can be overridden by the `rescale_factor` parameter in the `preprocess` method. - do_normalize: + do_normalize (`bool`, *optional*, defaults to `True`): Whether to normalize the image. Can be overridden by the `do_normalize` parameter in the `preprocess` method. image_mean (`float` or `List[float]`, *optional*, defaults to `IMAGENET_STANDARD_MEAN`):