Blame · requirements/requirements-test.txt · 62f4e2eb0760ac8bfe28834b061dbc2bda93ade9 · OpenDAS / ColossalAI · GitLab

Find file Normal view History Permalink

requirements-test.txt 268 Bytes

Newer Older

[tests] diffuser models in model zoo (#3136) HELSON committed Mar 14, 2023	1	`diffusers`
[CI] downgrade fbgemm. (#1778) Super Daniel committed Oct 31, 2022	2	`fbgemm-gpu==0.2.0`
Migrated project zbian committed Oct 28, 2021	3	`pytest`
[worfklow] added coverage test (#2399) Frank Lee committed Jan 09, 2023	4	`pytest-cov`
[dependency] removed torchvision (#833) Frank Lee committed Apr 22, 2022	5	`torchvision`
[hotfix] update requirements-test (#701) ver217 committed Apr 08, 2022	6	`transformers`
[fx] fixed unit tests for torch 1.12 (#1327) Frank Lee committed Jul 15, 2022	7	`timm`
[titans]remove model zoo (#1042) YuliangLiu0306 committed May 31, 2022	8	`titans`
[fx] add torchaudio test (#1369) Super Daniel committed Jul 27, 2022	9	`torchaudio`
[CI] downgrade fbgemm. (#1778) Super Daniel committed Oct 31, 2022	10	`torchrec==0.2.0`
[FAW] add cache manager for the cached embedding (#1419) Jiarui Fang committed Aug 09, 2022	11	`contexttimer`
[feat] add flash attention (#1762) oahzxl committed Oct 26, 2022	12	`einops`
[test] fixed the triton version for testing (#2608) Frank Lee committed Feb 07, 2023	13	`triton==2.0.0.dev20221202`
[CI] downgrade fbgemm. (#1778) Super Daniel committed Oct 31, 2022	14	`git+https://github.com/HazyResearch/flash-attention.git@c422fee3776eb3ea24e011ef641fd5fbeb212623#egg=flash_attn`