mirror of https://github.com/hpcaitech/ColossalAI
725af3eeeb
* feat: make optimizer optional in Booster.boost * test: skip unet test if diffusers version > 0.10.2 |
||
---|---|---|
.. | ||
benchmark_autochunk_diffuser.py | ||
test_autochunk_diffuser_utils.py | ||
test_autochunk_unet.py |