mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-04-30 20:04:42 +00:00
1216d1e7bd
* [tests] diffuser models in model zoo * remove useless code * [tests] add diffusers to requirement-test
15 lines
268 B
Plaintext
15 lines
268 B
Plaintext
diffusers
|
|
fbgemm-gpu==0.2.0
|
|
pytest
|
|
pytest-cov
|
|
torchvision
|
|
transformers
|
|
timm
|
|
titans
|
|
torchaudio
|
|
torchrec==0.2.0
|
|
contexttimer
|
|
einops
|
|
triton==2.0.0.dev20221202
|
|
git+https://github.com/HazyResearch/flash-attention.git@c422fee3776eb3ea24e011ef641fd5fbeb212623#egg=flash_attn
|