github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-24 06:29:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed a writing error

Browse Source
This commit is contained in:
yuehuayingxueluo 2024-01-02 18:34:19 +08:00 committed by FrankLeeeee
parent 9489dc64d8
commit 4df8876fca
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_infer/test_inference_engine.py
View File

@ -44,7 +44,7 @@ def run_dist(rank, world_size, port):
check_inference_engine(True) check_inference_engine(True)
check_inference_engine(False) check_inference_engine(False)
# TODO: There are some in sampler # TODO: There are some bugs in sampler.
# for s1, s2 in zip(cai_outputs, transformer_outputs): # for s1, s2 in zip(cai_outputs, transformer_outputs):
# assert s1 == s2 # assert s1 == s2