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
..
test_gemini_checkpoint_io.py	[devops] fix compatibility (#5444)	2024-03-13 15:24:13 +08:00
test_gemini_torch_compability.py	[workflow] fixed build CI (#5240)	2024-01-10 22:34:16 +08:00
test_general_checkpoint_io.py	[hotfix] fix lr scheduler bug in torch 2.0 (#4864)	2023-10-12 14:04:24 +08:00
test_hybrid_parallel_plugin_checkpoint_io.py	[hotfix] set return_outputs=False in examples and polish code (#5404)	2024-03-25 12:31:09 +08:00
test_low_level_zero_checkpoint_io.py	[Pipeline Inference] Sync pipeline inference branch to main (#4820)	2023-10-11 11:40:06 +08:00
test_plugins_huggingface_compatibility.py	[workflow] fixed build CI (#5240)	2024-01-10 22:34:16 +08:00
test_torch_ddp_checkpoint_io.py	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
test_torch_fsdp_checkpoint_io.py	[fsdp] impl save/load shard model/optimizer (#5357)	2024-02-27 13:51:14 +08:00
utils.py	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00