github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-08 20:40:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

[sync] resolve conflicts of merging main

Browse Source
This commit is contained in:
Yuanheng Zhao
2024-05-05 05:14:00 +00:00
parent 537a3cbc4d d3f34ee8cc
commit 56ed09aba5
348 changed files with 4639 additions and 2395 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/language/openmoe/benchmark/benchmark_cai.py
View File

@@ -146,7 +146,7 @@ def main():
args = parse_args()
# Launch ColossalAI
colossalai.launch_from_torch(config={}, seed=args.seed)
colossalai.launch_from_torch(seed=args.seed)
coordinator = DistCoordinator()
# Set plugin

2
examples/language/openmoe/train.py
View File

@@ -207,7 +207,7 @@ def main():
args = parse_args()
# Launch ColossalAI
colossalai.launch_from_torch(config={}, seed=args.seed)
colossalai.launch_from_torch(seed=args.seed)
coordinator = DistCoordinator()
test_mode = args.model_name == "test"