robert commited on
Commit
146bc3a
·
1 Parent(s): 6b0660d

Setting cuda version explicitly

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -0
requirements.txt CHANGED
@@ -1,3 +1,5 @@
 
 
1
  langchain-community==0.2.7
2
  langchain-core==0.2.12
3
  langchain-huggingface==0.0.3
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu118
2
+ torch==2.0.1
3
  langchain-community==0.2.7
4
  langchain-core==0.2.12
5
  langchain-huggingface==0.0.3