chore(requirements) add torchvision

neurocode-ai · Sep 10, 2023 · fbe6c9d · fbe6c9d
1 parent d22e82a
commit fbe6c9d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -5,4 +5,5 @@ pandas
 tqdm
 scikit-learn
 umap-learn
-torch>=1.7.0+cpu
+torch>=1.7.0+cpu
+torchvision