mirror of https://github.com/hpcaitech/ColossalAI
![]() * fix * padding vocab_size when using pipeline parallellism padding vocab_size when using pipeline parallellism fix fix * fix * fix fix fix * fix gather output * fix * fix * fix fix resize embedding fix resize embedding * fix resize embedding fix * revert * revert * revert * fix lm forward distribution * fix * test ci * fix |
||
---|---|---|
.. | ||
_C | ||
_analyzer | ||
accelerator | ||
amp | ||
auto_parallel | ||
autochunk | ||
booster | ||
checkpoint_io | ||
cli | ||
cluster | ||
context | ||
device | ||
fx | ||
inference | ||
interface | ||
kernel | ||
lazy | ||
legacy | ||
logging | ||
moe | ||
nn | ||
pipeline | ||
shardformer | ||
tensor | ||
testing | ||
utils | ||
zero | ||
__init__.py | ||
initialize.py |