2023-03-14 09:20:28 +00:00
diffusers
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-09-09 14:45:36 +00:00
transformers==4.33.0
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-11-16 08:43:15 +00:00
triton==2.1.0
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
2023-12-27 10:03:45 +00:00
flash_attn
2023-07-20 02:39:06 +00:00
datasets
2023-10-30 02:52:19 +00:00
pydantic
ray
2023-09-22 03:02:50 +00:00
#auto-gptq now not support torch1.12