main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\legacy\zero\sharded_model\sharded_model_v2.py
114 1 1
A
colossalai\shardformer\modeling\whisper.py
112 2 4
A
colossalai\zero\gemini\chunk\chunk.py
112 7 -
A
examples\tutorial\sequence_parallel\data\tokenizer\bert_tokenization.py
111 1 1
A
examples\images\diffusion\ldm\modules\diffusionmodules\model.py
111 1 2
A
examples\images\diffusion\ldm\models\diffusion\dpm_solver\dpm_solver.py
110 1 1
A
colossalai\checkpoint_io\utils.py
109 2 -
A
examples\tutorial\sequence_parallel\data\datasets\indexed_dataset.py
107 1 -
A
examples\images\diffusion\ldm\modules\image_degradation\bsrgan.py
105 1 2
A
applications\Chat\examples\community\ray\train_prompts_on_ray.py
104 1 -
A
examples\images\diffusion\ldm\modules\diffusionmodules\openaimodel.py
101 1 1
A
colossalai\shardformer\modeling\bloom.py
101 2 3
A
colossalai\shardformer\modeling\t5.py
99 1 4
A
colossalai\shardformer\modeling\gpt2.py
99 2 2
A
colossalai\legacy\nn\parallel\layers\cache_embedding\cache_mgr.py
95 1 1
A
colossalai\shardformer\modeling\opt.py
95 2 2
A
colossalai\inference\quant\smoothquant\models\llama.py
94 4 3
A
colossalai\kernel\cuda_native\csrc\scaled_upper_triang_masked_softmax.h
94 1 4
A
colossalai\kernel\cuda_native\csrc\scaled_masked_softmax.h
93 1 4
A
examples\language\openmoe\model\openmoe_policy.py
93 1 1