"tests/vscode:/vscode.git/clone" did not exist on "a04ebc8b33314c42349c3e12885960a292c9c9dd"
Unverified Commit 70ba10e6 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Fix `LeViT` checkpoint (#19069)


Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent bc5d0b10
......@@ -37,7 +37,7 @@ class LevitConfig(PretrainedConfig):
This is the configuration class to store the configuration of a [`LevitModel`]. It is used to instantiate a LeViT
model according to the specified arguments, defining the model architecture. Instantiating a configuration with the
defaults will yield a similar configuration to that of the LeViT
[facebook/levit-base-192](https://huggingface.co/facebook/levit-base-192) architecture.
[facebook/levit-128S](https://huggingface.co/facebook/levit-128S) architecture.
Configuration objects inherit from [`PretrainedConfig`] and can be used to control the model outputs. Read the
documentation from [`PretrainedConfig`] for more information.
......@@ -78,10 +78,10 @@ class LevitConfig(PretrainedConfig):
```python
>>> from transformers import LevitModel, LevitConfig
>>> # Initializing a LeViT levit-base-192 style configuration
>>> # Initializing a LeViT levit-128S style configuration
>>> configuration = LevitConfig()
>>> # Initializing a model from the levit-base-192 style configuration
>>> # Initializing a model from the levit-128S style configuration
>>> model = LevitModel(configuration)
>>> # Accessing the model configuration
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment