riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

Fix LeViT checkpoint (#19069)

Browse Source 
Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar 2022-09-16 16:23:58 +02:00 committed by GitHub
parent bc5d0b1046
commit 70ba10e6d4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/transformers/models/levit/configuration_levit.py
View File

@ -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