Skip to content

Commit 00efd64

Browse files
authored
Fix vilt config docstring parameter to match value in init (#26017)
* Fix vilt config init parameter to match the ones in documentation * Fix the documentation
1 parent 02c4a77 commit 00efd64

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/transformers/models/vilt/configuration_vilt.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -58,9 +58,9 @@ class ViltConfig(PretrainedConfig):
5858
hidden_act (`str` or `function`, *optional*, defaults to `"gelu"`):
5959
The non-linear activation function (function or string) in the encoder and pooler. If string, `"gelu"`,
6060
`"relu"`, `"selu"` and `"gelu_new"` are supported.
61-
hidden_dropout_prob (`float`, *optional*, defaults to 0.1):
61+
hidden_dropout_prob (`float`, *optional*, defaults to 0.0):
6262
The dropout probabilitiy for all fully connected layers in the embeddings, encoder, and pooler.
63-
attention_probs_dropout_prob (`float`, *optional*, defaults to 0.1):
63+
attention_probs_dropout_prob (`float`, *optional*, defaults to 0.0):
6464
The dropout ratio for the attention probabilities.
6565
initializer_range (`float`, *optional*, defaults to 0.02):
6666
The standard deviation of the truncated_normal_initializer for initializing all weight matrices.

0 commit comments

Comments
 (0)