From e7a64e5af81bd25eab5bc690ca4464657c9493c1 Mon Sep 17 00:00:00 2001
From: Mauricio Villegas <5780272+mauvilsa@users.noreply.github.com>
Date: Tue, 9 Jul 2024 20:19:54 +0200
Subject: [PATCH] Update modeling_utils.py

Add return type annotation to PreTrainedModel.from_pretrained
---
 src/transformers/modeling_utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/transformers/modeling_utils.py b/src/transformers/modeling_utils.py
index 4d59635c90e97b..b37f968ae92625 100755
--- 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.