Update Wall-X to 1.1.0 (#104)

2026-06-15 11:40:00 +08:00
parent e23a586846
commit 72834e7de5
200 changed files with 33916 additions and 16771 deletions
@@ -0,0 +1,34 @@
+"""Distributed trainer package exports."""
+
+from wall_x.trainer.fsdp_trainer.base_trainer import (
+    BaseDistributedTrainer,
+    DistributedTrainer,
+    all_gather,
+    all_reduce,
+    barrier,
+    cleanup_distributed,
+    get_local_rank,
+    get_rank,
+    get_world_size,
+    is_main_process,
+    setup_distributed,
+)
+from wall_x.trainer.fsdp_trainer.fsdp_trainer import FSDPTrainer
+
+__all__ = [
+    # Base classes and utilities
+    "BaseDistributedTrainer",
+    "DistributedTrainer",
+    "setup_distributed",
+    "cleanup_distributed",
+    "get_rank",
+    "get_world_size",
+    "get_local_rank",
+    "is_main_process",
+    "barrier",
+    "all_reduce",
+    "all_gather",
+    # Trainers
+    # "DDPTrainer",
+    "FSDPTrainer",
+]