ColossalAI/tests/test_checkpoint_io
flybird11111 8954a0c2e2 [LowLevelZero] low level zero support lora (#5153)
* low level zero support lora

low level zero support lora

* add checkpoint test

* add checkpoint test

* fix

* fix

* fix

* fix

fix

fix

fix

* fix

* fix

fix

fix

fix

fix

fix

fix

* fix

* fix

fix

fix

fix

fix

fix

fix

* fix

* test ci

* git # This is a combination of 3 commits.

Update low_level_zero_plugin.py

Update low_level_zero_plugin.py

fix

fix

fix

* fix naming

fix naming

fix naming

fix
2024-04-28 10:51:27 +08:00
..
test_gemini_checkpoint_io.py [shardformer] Sequence Parallelism Optimization (#5533) 2024-04-03 17:15:47 +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 [shardformer] refactor embedding resize (#5603) 2024-04-18 16:10:18 +08:00
test_low_level_zero_checkpoint_io.py [LowLevelZero] low level zero support lora (#5153) 2024-04-28 10:51:27 +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 [devops] remove post commit ci (#5566) 2024-04-08 15:09:40 +08:00
utils.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00