2023-03-14 09:20:28 +00:00
diffusers
2022-10-31 10:18:45 +00:00
fbgemm-gpu==0.2.0
2021-10-28 16:21:23 +00:00
pytest
2023-05-24 06:59:40 +00:00
coverage==7.2.3
2023-05-15 09:20:56 +00:00
git+https://github.com/hpcaitech/pytest-testmon
2022-04-22 07:24:35 +00:00
torchvision
2023-07-19 08:43:01 +00:00
transformers==4.30.2
2022-07-15 10:22:15 +00:00
timm
2022-05-31 02:40:47 +00:00
titans
2022-07-27 03:03:14 +00:00
torchaudio
2023-05-15 09:20:56 +00:00
torchx-nightly==2022.6.29 # torchrec 0.2.0 requires torchx-nightly. This package is updated every day. We fix the version to a specific date to avoid breaking changes.
2022-10-31 10:18:45 +00:00
torchrec==0.2.0
2022-08-09 07:17:17 +00:00
contexttimer
2022-10-26 08:15:52 +00:00
einops
2023-02-07 05:49:38 +00:00
triton==2.0.0.dev20221202
2023-04-06 06:51:35 +00:00
requests==2.27.1 # downgrade to avoid huggingface error https://github.com/huggingface/transformers/issues/17611
2023-06-15 08:50:08 +00:00
SentencePiece
2023-08-09 06:24:45 +00:00
ninja
flash_attn>=2.0
2023-07-20 02:39:06 +00:00
datasets