Skip to content

Commit

Permalink
fix docstrings
Browse files Browse the repository at this point in the history
  • Loading branch information
echarlaix committed Jul 12, 2024
1 parent f755a58 commit f351326
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions optimum/modeling_base.py
Original file line number Diff line number Diff line change
Expand Up @@ -47,8 +47,8 @@
user or organization name, like `dbmdz/bert-base-german-cased`.
- A path to a *directory* containing a model saved using [`~OptimizedModel.save_pretrained`],
e.g., `./my_model_directory/`.
from_transformers (`bool`, defaults to `False`):
Defines whether the provided `model_id` contains a vanilla Transformers checkpoint.
export (`bool`, defaults to `False`):
Defines whether the provided `model_id` needs to be exported to the targeted format.
force_download (`bool`, defaults to `True`):
Whether or not to force the (re-)download of the model weights and configuration files, overriding the
cached versions if they exist.
Expand All @@ -71,6 +71,10 @@
Whether or not to allow for custom code defined on the Hub in their own modeling. This option should only be set
to `True` for repositories you trust and in which you have read the code, as it will execute code present on
the Hub on your local machine.
revision (`Optional[str]`, defaults to `None`):
The specific model version to use. It can be a branch name, a tag name, or a commit id, since we use a
git-based system for storing models and other artifacts on huggingface.co, so `revision` can be any
identifier allowed by git.
"""


Expand Down

0 comments on commit f351326

Please sign in to comment.