From 73098f2e6b2a22935535edaab7273a15d763a628 Mon Sep 17 00:00:00 2001
From: Hecate0821 <Hec4te0821@gmail.com>
Date: Wed, 7 Jan 2026 10:33:46 +0000
Subject: [PATCH] Fix import

---
 slime/backends/fsdp_utils/__init__.py | 7 +------
 1 file changed, 1 insertion(+), 6 deletions(-)

diff --git a/slime/backends/fsdp_utils/__init__.py b/slime/backends/fsdp_utils/__init__.py
index 05099e93c..444b43089 100644
--- a/slime/backends/fsdp_utils/__init__.py
+++ b/slime/backends/fsdp_utils/__init__.py
@@ -1,13 +1,8 @@
 import logging
 
-
 try:
-    _TORCH_MEMORY_SAVER_AVAILABLE = True
-except ImportError:
-    logging.warning("torch_memory_saver is not installed, refer to : https://github.com/fzyzcjy/torch_memory_saver")
-    _TORCH_MEMORY_SAVER_AVAILABLE = False
+    from torch.distributed.fsdp import fully_shard  # noqa: F401
 
-try:
     _FSDP_AVAILABLE = True
 except ImportError as e:
     logging.warning(f"FSDP backend dependencies not available: {e}")