-
Notifications
You must be signed in to change notification settings - Fork 487
Closed
Description
slime/slime/backends/fsdp_utils/__init__.py
Lines 4 to 14 in 0ab4151
| 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 |
I wonder the purpose of the quoted code and where the import actually gets checked.
Metadata
Metadata
Assignees
Labels
No labels