From 0ee1bb050778c1e7864e66e6359cd2a9492a0a06 Mon Sep 17 00:00:00 2001
From: Xu Zhao <xzhao9@meta.com>
Date: Mon, 17 Jun 2024 15:14:49 -0400
Subject: [PATCH] Remove torchvision and torchaudio dependency when testing
 userbenchmark.

---
 install.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/install.py b/install.py
index ea825e6f7c..943de9dc11 100644
--- a/install.py
+++ b/install.py
@@ -55,6 +55,8 @@ def pip_install_requirements(requirements_txt="requirements.txt"):
         end="",
         flush=True,
     )
+    if args.userbenchmark:
+        TORCH_DEPS = ["torch"]
     try:
         versions = get_pkg_versions(TORCH_DEPS)
     except ModuleNotFoundError as e: