diff --git a/requirements/requirements-test.txt b/requirements/requirements-test.txt
index 380a3f3bf..6eba3984d 100644
--- a/requirements/requirements-test.txt
+++ b/requirements/requirements-test.txt
@@ -1,12 +1,13 @@
 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