Bug fix

xuzhao9 · xuzhao9 · commit 7c98b1c971f6 · 2024-06-19T19:22:31.000-04:00
diff --git a/torchbenchmark/models/pytorch_unet/install.py b/torchbenchmark/models/pytorch_unet/install.py
@@ -1,4 +1,4 @@
 from utils.python_utils import pip_install_requirements
 
 if __name__ == '__main__':
-    pip_install_requirements()
+    pip_install_requirements(requirements_txt="pytorch_unet/requirements.txt")
