ColossalAI

History

yuehuayingxueluo e8f0642f28 [Inference]Add Nopadding Llama Modeling (#5327 ) * add nopadding llama modeling * add nopadding_llama.py * rm unused codes * fix bugs in test_xine_copy.py * fix code style		10 months ago
..
_C	…
_analyzer	…
amp	[npu] add npu support for gemini and zero (#5067 )	1 year ago
auto_parallel	[npu] add npu support for gemini and zero (#5067 )	1 year ago
autochunk	…
booster	[ci] fix shardformer tests. (#5255 )	11 months ago
checkpoint_io	[pipeline,shardformer] Fix p2p efficiency in pipeline, allow skipping loading weight not in weight_map when `strict=False`, fix llama flash attention forward, add flop estimation by megatron in llama benchmark (#5017 )	1 year ago
cli	[bug] Fix the version check bug in colossalai run when generating the cmd. (#4713 )	1 year ago
cluster	fix-test (#5210 )	11 months ago
context	[moe] merge moe into main (#4978 )	1 year ago
device	[npu] add npu support for hybrid plugin and llama (#5090 )	1 year ago
fx	…
inference	[Inference]Add Nopadding Llama Modeling (#5327 )	10 months ago
interface	[lazy] support from_pretrained (#4801 )	1 year ago
kernel	[Inference] Update rms norm kernel, benchmark with vLLM (#5315 )	10 months ago
lazy	[doc] add lazy init docs (#4808 )	1 year ago
legacy	adapted to pad_context_forward	11 months ago
logging	…
moe	[hotfix]: modify create_ep_hierarchical_group and add test (#5032 )	1 year ago
nn	[npu] add npu support for gemini and zero (#5067 )	1 year ago
pipeline	[pipeline] A more general _communicate in p2p (#5062 )	11 months ago
shardformer	[doc] fix doc typo (#5256 )	11 months ago
tensor	fix (#5158 )	12 months ago
testing	[npu] add npu support for hybrid plugin and llama (#5090 )	1 year ago
utils	[shardformer]: support gpt-j, falcon, Mistral and add interleaved pipeline for bert (#5088 )	1 year ago
zero	[shardformer]: support gpt-j, falcon, Mistral and add interleaved pipeline for bert (#5088 )	1 year ago
__init__.py	…
initialize.py	[npu] add npu support for gemini and zero (#5067 )	1 year ago