From f28f39792d52b100e9829f5f3210b17d68b12635 Mon Sep 17 00:00:00 2001
From: SamAcctX <87765660+SamAcctX@users.noreply.github.com>
Date: Sun, 2 Feb 2025 17:41:36 -0600
Subject: [PATCH] update deprecated deepspeed import for transformers 4.46+
 (#6725)

---
 modules/models.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/modules/models.py b/modules/models.py
index f551b828..3951fe82 100644
--- a/modules/models.py
+++ b/modules/models.py
@@ -33,7 +33,7 @@ transformers.logging.set_verbosity_error()
 local_rank = None
 if shared.args.deepspeed:
     import deepspeed
-    from transformers.deepspeed import (
+    from transformers.integrations.deepspeed import (
         HfDeepSpeedConfig,
         is_deepspeed_zero3_enabled
     )