[CI] downgrade fbgemm. (#1778)

5ea89f64 · Super Daniel · GitHub · f34dab42 · 5ea89f64
Unverified Commit 5ea89f64 authored Oct 31, 2022 by Super Daniel Committed by GitHub Oct 31, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 2 deletions

requirements/requirements-test.txt requirements/requirements-test.txt +3 -2

No files found.
--- a/requirements/requirements-test.txt
+++ b/requirements/requirements-test.txt
 diffusers
+fbgemm-gpu==0.2.0
 pytest
 torchvision
 transformers
 timm
 titans
 torchaudio
-torchrec
+torchrec==0.2.0
 contexttimer
 einops
 triton==2.0.0.dev20221011
-git+https://github.com/HazyResearch/flash-attention.git@c422fee3776eb3ea24e011ef641fd5fbeb212623#egg=flash_attn
\ No newline at end of file
+git+https://github.com/HazyResearch/flash-attention.git@c422fee3776eb3ea24e011ef641fd5fbeb212623#egg=flash_attn