ColossalAI/colossalai/shardformer/policies
littsk 11f1e426fe
[hotfix] Correct several erroneous code comments (#4794)
2023-09-27 10:43:03 +08:00
..
__init__.py [shardformer] init shardformer code structure (#3731) 2023-07-04 16:05:01 +08:00
auto_policy.py [inference] chatglm2 infer demo (#4724) 2023-09-22 11:12:50 +08:00
base_policy.py [hotfix] Correct several erroneous code comments (#4794) 2023-09-27 10:43:03 +08:00
bert.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
blip2.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
bloom.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
chatglm2.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
gpt2.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
llama.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
opt.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
sam.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
t5.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
vit.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
whisper.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00