botbw d097224d90 [feat] support qwen3 in shardformer

2025-07-10 13:57:52 +08:00

..

__init__.py

…

auto_policy.py

[feat] support qwen3 in shardformer

2025-07-10 13:57:52 +08:00

base_policy.py

[fp8] Merge feature/fp8_comm to main branch of Colossalai (#6016)

2024-08-22 09:21:34 +08:00

bert.py

[upgrade]Upgrade transformers (#6320)

2025-05-27 14:29:01 +08:00

blip2.py

[Sharderformer] Support zbv in Sharderformer Policy (#6150)

2025-01-02 10:22:26 +08:00

bloom.py

[upgrade]Upgrade transformers (#6320)

2025-05-27 14:29:01 +08:00

chatglm2.py

[Sharderformer] Support zbv in Sharderformer Policy (#6150)

2025-01-02 10:22:26 +08:00

command.py

[upgrade]Upgrade transformers (#6320)

2025-05-27 14:29:01 +08:00

deepseek_v3.py

[shardformer] support pipeline for deepseek v3 and optimize lora save (#6188)

2025-02-14 14:48:54 +08:00

deepseek.py

[Sharderformer] Support zbv in Sharderformer Policy (#6150)

2025-01-02 10:22:26 +08:00

falcon.py

[upgrade]Upgrade transformers (#6320)

2025-05-27 14:29:01 +08:00

gpt2.py

[upgrade]Upgrade transformers (#6320)

2025-05-27 14:29:01 +08:00

gptj.py

[Sharderformer] Support zbv in Sharderformer Policy (#6150)

2025-01-02 10:22:26 +08:00

llama.py

[upgrade]Upgrade transformers (#6320)

2025-05-27 14:29:01 +08:00

mistral.py

[upgrade]Upgrade transformers (#6320)

2025-05-27 14:29:01 +08:00

mixtral.py

[upgrade]Upgrade transformers (#6320)

2025-05-27 14:29:01 +08:00

opt.py

[Sharderformer] Support zbv in Sharderformer Policy (#6150)

2025-01-02 10:22:26 +08:00

qwen2.py

[feat] support qwen3 in shardformer

2025-07-10 13:57:52 +08:00

qwen3.py

[feat] support qwen3 in shardformer

2025-07-10 13:57:52 +08:00

sam.py

[Sharderformer] Support zbv in Sharderformer Policy (#6150)

2025-01-02 10:22:26 +08:00

t5.py

[Sharderformer] Support zbv in Sharderformer Policy (#6150)

2025-01-02 10:22:26 +08:00

vit.py

[upgrade]Upgrade transformers (#6320)

2025-05-27 14:29:01 +08:00

whisper.py

[Sharderformer] Support zbv in Sharderformer Policy (#6150)

2025-01-02 10:22:26 +08:00