ColossalAI/examples/language/palm/palm_pytorch
ZijianYY 92de90dfb3
[examples] replace einsum with matmul (#2210)
2022-12-28 19:03:06 +08:00
..
__init__.py [example] add palm pytorch version (#2172) 2022-12-22 10:15:34 +08:00
autoregressive_wrapper.py [example] add palm pytorch version (#2172) 2022-12-22 10:15:34 +08:00
palm_pytorch.py [examples] replace einsum with matmul (#2210) 2022-12-28 19:03:06 +08:00