trending/miles
1
0
Fork 0
mirror of https://github.com/radixark/miles.git synced 2026-10-02 07:14:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

Remove eager FSDP package imports to break configuration cycles

Browse Source 
Materialize M48 G02 from the frozen revision-003 patch and source map 86efe0f5. Preserve the reviewed patch boundary and keep unit-test migrations in the final GU operation.
This commit is contained in:
Tom
2026-10-01 15:43:35 +08:00
parent de6766031e
commit ef8a22a475
1 changed files with 0 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files
miles/backends/fsdp_utils/__init__.py
-33
View File
@@ -1,33 +0,0 @@
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
try:
_FSDP_AVAILABLE = True
except ImportError as e:
logging.warning(f"FSDP backend dependencies not available: {e}")
_FSDP_AVAILABLE = False
if _FSDP_AVAILABLE:
from .actor import FSDPTrainRayActor
from .arguments import load_fsdp_args
else:
def _raise_import_error(*args, **kwargs):
raise ImportError(
"FSDP backend is not available. "
"Please ensure PyTorch with FSDP2 support is installed. "
"For installation instructions, refer to: https://pytorch.org/docs/stable/distributed.fsdp.fully_shard.html"
)
FSDPTrainRayActor = _raise_import_error
load_fsdp_args = _raise_import_error
__all__ = ["load_fsdp_args", "FSDPTrainRayActor"]
logging.getLogger().setLevel(logging.WARNING)