ViperEkura/AstrAI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

refactor: inline parallel utils into executor module

Browse Source 
- Move create_ref_model from astrai/parallel/utils.py into executor.py
- Remove unused ColumnParallelLinear/RowParallelLinear (module.py)
- Update imports in strategy.py and train_context.py
- Drop unused astrai.parallel.utils and astrai.parallel.module
This commit is contained in:
ViperEkura
2026-07-30 07:54:54 +08:00
parent c8b1e40f71
commit 6a97524db4
6 changed files with 28 additions and 157 deletions
Download Patch File Download Diff File Expand all files Collapse all files
astrai/trainer/strategy.py
-1
View File
@@ -9,7 +9,6 @@ import torch.nn.functional as F
from torch import Tensor
from astrai.factory import BaseFactory
from astrai.parallel.utils import create_ref_model
from astrai.trainer.rollout import RolloutResult