From 053e974bbfe33c2b5cd559aa26938fc0c537bf0e Mon Sep 17 00:00:00 2001
From: PaulZhang12 <paulzhan@umich.edu>
Date: Fri, 27 Sep 2024 19:43:35 +0000
Subject: [PATCH] Update requirements

---
 .ci/docker/requirements.txt | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/.ci/docker/requirements.txt b/.ci/docker/requirements.txt
index 2384fb1b00..b5a7448050 100644
--- a/.ci/docker/requirements.txt
+++ b/.ci/docker/requirements.txt
@@ -70,3 +70,5 @@ pycocotools
 semilearn==0.3.2
 torchao==0.0.3
 segment_anything==1.0
+torchrec
+fbgemm-gpu
\ No newline at end of file