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

Update modeling_utils.py Add return type annotation to PreTrainedModel.from_pretrained
huggingface · Jul 9, 2024 · c5bc2d5 · c5bc2d5
1 parent 6e59b30
commit c5bc2d5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/modeling_utils.py b/src/transformers/modeling_utils.py
@@ -2832,7 +2832,7 @@ def from_pretrained(
         revision: str = "main",
         use_safetensors: bool = None,
         **kwargs,
-    ):
+    ) -> "PreTrainedModel":
         r"""
         Instantiate a pretrained pytorch model from a pre-trained model configuration.