ColossalAI/tests/test_autochunk/test_autochunk_diffuser
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
..
benchmark_autochunk_diffuser.py [autochunk] support complete benchmark (#3121) 2023-03-13 17:42:37 +08:00
test_autochunk_diffuser_utils.py [devops] update torch version of CI (#3725) 2023-05-15 17:20:56 +08:00
test_autochunk_unet.py [booster] make optimizer argument optional for boost (#3993) 2023-06-15 17:38:42 +08:00