You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

GuangyaoZhang d84d68601a change 'xxx if xxx else None' to 'xxx or None'		5 months ago
..
__init__.py	…
auto_policy.py	change command	5 months ago
base_policy.py	[shardformer] refactor pipeline grad ckpt config (#5646)	7 months ago
bert.py	change 'xxx if xxx else None' to 'xxx or None'	5 months ago
blip2.py	[Shardformer] add assert for num of attention heads divisible by tp_size (#5670)	7 months ago
bloom.py	change 'xxx if xxx else None' to 'xxx or None'	5 months ago
chatglm2.py	change 'xxx if xxx else None' to 'xxx or None'	5 months ago
command.py	change 'xxx if xxx else None' to 'xxx or None'	5 months ago
falcon.py	[Shardformer] Add parallel output for shardformer models(bloom, falcon) (#5702)	6 months ago
gpt2.py	change 'xxx if xxx else None' to 'xxx or None'	5 months ago
gptj.py	[shardformer] upgrade transformers to 4.39.3 (#5815)	6 months ago
llama.py	change 'xxx if xxx else None' to 'xxx or None'	5 months ago
mistral.py	[shardformer] upgrade transformers to 4.39.3 (#5815)	6 months ago
opt.py	[pre-commit.ci] auto fixes from pre-commit.com hooks	7 months ago
qwen2.py	[Shardformer]fix the num_heads assert for llama model and qwen model (#5704)	7 months ago
sam.py	[Shardformer] add assert for num of attention heads divisible by tp_size (#5670)	7 months ago
t5.py	[Shardformer] add assert for num of attention heads divisible by tp_size (#5670)	7 months ago
vit.py	[Shardformer] add assert for num of attention heads divisible by tp_size (#5670)	7 months ago
whisper.py	[Shardformer] add assert for num of attention heads divisible by tp_size (#5670)	7 months ago