ColossalAI/tests/kit/model_zoo/transformers
flybird11111 e1c72fd41c fix
2025-05-22 16:49:06 +08:00
..
__init__.py [release] update version (#6195) 2025-02-20 11:36:18 +08:00
albert.py
bert.py fix 2025-05-22 16:49:06 +08:00
blip2.py
bloom.py
chatglm2.py
command.py [Feature] Zigzag Ring attention (#5905) 2024-08-16 13:56:38 +08:00
deepseek_v3.py [release] update version (#6195) 2025-02-20 11:36:18 +08:00
deepseek.py [misc] remove debug/print code 2024-08-01 10:06:59 +08:00
falcon.py
gpt.py [Feature] Split cross-entropy computation in SP (#5959) 2024-09-10 12:06:50 +08:00
gptj.py
llama.py [Feature] Zigzag Ring attention (#5905) 2024-08-16 13:56:38 +08:00
mistral.py [Feature] Zigzag Ring attention (#5905) 2024-08-16 13:56:38 +08:00
mixtral.py
opt.py [upgrade]upgrade opt (#6307) 2025-05-21 16:13:32 +08:00
qwen2.py [Feature] Zigzag Ring attention (#5905) 2024-08-16 13:56:38 +08:00
sam.py
t5.py
vit.py
whisper.py