Unverified Commit c5bc2d5f authored by Mauricio Villegas's avatar Mauricio Villegas Committed by GitHub
Browse files

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

Update modeling_utils.py

Add return type annotation to PreTrainedModel.from_pretrained
parent 6e59b308
......@@ -2832,7 +2832,7 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin, GenerationMixin, PushToHubMix
revision: str = "main",
use_safetensors: bool = None,
**kwargs,
):
) -> "PreTrainedModel":
r"""
Instantiate a pretrained pytorch model from a pre-trained 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