ColossalAI/tests/test_autochunk
Wenhao Chen 725af3eeeb
[booster] make optimizer argument optional for boost (#3993)
* feat: make optimizer optional in Booster.boost

* test: skip unet test if diffusers version > 0.10.2
2023-06-15 17:38:42 +08:00
..
test_autochunk_alphafold [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_autochunk_diffuser [booster] make optimizer argument optional for boost (#3993) 2023-06-15 17:38:42 +08:00
test_autochunk_transformer [devops] update torch version of CI (#3725) 2023-05-15 17:20:56 +08:00
test_autochunk_vit [devops] update torch version of CI (#3725) 2023-05-15 17:20:56 +08:00