ColossalAI/tests/test_tensor/model/test_gpt2.py at af718e83f2d1db2c9500a76c16128fe109ab44ac - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-13 20:39:59 +00:00

Files

HELSON 7a065dc9f6 [hotfix] fix megatron_init in test_gpt2.py (#1357 )

2022-07-25 10:28:19 +08:00

5.6 KiB

Raw Blame History

View Raw