chore : 升级 torch 2.11.0+cu128,移除自定义 Muon,修复 gloo device_id

- torch 2.7.1-cu126 升级至 2.11.0-cu128,numpy 2.3.2 升级至 2.4.4
- 移除 astrai/trainer/optim.py,改用 torch.optim.Muon
- parallel setup: gloo 后端不再传递 device_id,单卡多进程不再报错
This commit is contained in:
2026-06-27 16:10:37 +08:00
parent b4587c5d08
commit 6715461a36
6 changed files with 9 additions and 155 deletions
-2
View File
@@ -47,7 +47,6 @@ from astrai.trainer import (
BaseScheduler,
BaseStrategy,
CallbackFactory,
Muon,
SchedulerFactory,
StrategyFactory,
TrainCallback,
@@ -75,7 +74,6 @@ __all__ = [
"GenerationRequest",
"InferenceEngine",
"LoRAConfig",
"Muon",
"Pipeline",
"PipelineConfig",
"ProtocolHandler",