mirror of https://github.com/hpcaitech/ColossalAI
8e718a1421
* [gemini] fix missing return * [gemini] fix missing arg pass * [gemini] use gather tensor instead of list * [test] enable flash attention for benchmark by default * [test] enable flash attention for benchmark by default --------- Co-authored-by: genghaozhe <939857490@qq.com> |
||
|---|---|---|
| .. | ||
| bert | ||
| commons | ||
| gpt | ||
| grok-1 | ||
| llama | ||
| openmoe | ||
| opt | ||
| palm | ||
| __init__.py | ||
| data_utils.py | ||
| model_utils.py | ||
| performance_evaluator.py | ||