github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-05 12:24:38 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
d5eeeb1416e6c4241026ce96480fb24395c1db4c
ColossalAI/colossalai/testing/utils.py
Xuanlei Zhao 3acbf6d496 [npu] add npu support for hybrid plugin and llama (#5090) 
* llama 3d

* update

* fix autocast
2023-11-22 19:23:21 +08:00

9.0 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink