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_autochunk_diffuser [booster] make optimizer argument optional for boost (#3993) 2023-06-15 17:38:42 +08:00
test_autochunk_transformer
test_autochunk_vit