add torch requirement

PreferredAI · Oct 20, 2023 · 7583783 · 7583783
1 parent 17ec685
commit 7583783
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/cornac/models/ncf/requirements.txt b/cornac/models/ncf/requirements.txt
@@ -1 +1,2 @@
-tensorflow==2.12.0
+tensorflow==2.12.0
+torch>=0.4.1