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

Add return type annotation to PreTrainedModel.from_pretrained (#31869)

Browse Source 
Update modeling_utils.py

Add return type annotation to PreTrainedModel.from_pretrained
This commit is contained in:
Mauricio Villegas 2024-07-09 22:49:29 +02:00 committed by GitHub
parent 6e59b30841
commit c5bc2d5fd5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/modeling_utils.py
View File

@ -2832,7 +2832,7 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin, GenerationMixin, PushToHubMix
revision: str = "main", revision: str = "main",
use_safetensors: bool = None, use_safetensors: bool = None,
**kwargs, **kwargs,
): ) -> "PreTrainedModel":
r""" r"""
Instantiate a pretrained pytorch model from a pre-trained model configuration. Instantiate a pretrained pytorch model from a pre-trained model configuration.