Wenhao Chen bb0a668fee [hotfix] set return_outputs=False in examples and polish code (#5404) ... * fix: simplify merge_batch * fix: use return_outputs=False to eliminate extra memory consumption * feat: add return_outputs warning * style: remove `return_outputs=False` as it is the default value		2024-03-25 12:31:09 +08:00
..
benchmark_cai.py	[hotfix] set return_outputs=False in examples and polish code (#5404)	2024-03-25 12:31:09 +08:00
benchmark_cai.sh	[moe] merge moe into main (#4978)	2023-11-02 02:21:24 +00:00
benchmark_cai_dist.sh	[moe] merge moe into main (#4978)	2023-11-02 02:21:24 +00:00
benchmark_fsdp.py	[hotfix] fix typo of openmoe model source (#5403)	2024-03-05 21:44:38 +08:00
benchmark_fsdp.sh	[moe] merge moe into main (#4978)	2023-11-02 02:21:24 +00:00
hostfile.txt	[moe] merge moe into main (#4978)	2023-11-02 02:21:24 +00:00
utils.py	[moe] merge moe into main (#4978)	2023-11-02 02:21:24 +00:00