github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Fixed a writing error

pull/5258/head
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