From 6ea05b0e34b92f595a6a2dcc684df0e8fd1d1001 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ye=20Canming=20=E5=8F=B6=E7=92=A8=E9=93=AD?= <41530341+2catycm@users.noreply.github.com> Date: Tue, 3 Dec 2024 16:07:26 +0800 Subject: [PATCH] feat(basemodel.py): Remove unused import that harms compatibility removed `transformers.deepspeed` since in the latest version like 4.46.3 will raise `ModuleNotFoundError: No module named 'transformers.deepspeed` causing running bugs. And it is totally not used by opendelta. --- opendelta/basemodel.py | 1 - 1 file changed, 1 deletion(-) diff --git a/opendelta/basemodel.py b/opendelta/basemodel.py index 1fb60a6..39b73ff 100644 --- a/opendelta/basemodel.py +++ b/opendelta/basemodel.py @@ -17,7 +17,6 @@ from opendelta.utils.decorate import decorate from opendelta.utils.structure_mapping import transform from transformers.file_utils import PushToHubMixin -from transformers.deepspeed import deepspeed_config, is_deepspeed_zero3_enabled from opendelta import SaveLoadMixin from opendelta import logging from opendelta.utils.structure_mapping import CommonStructureMap