Skip to content

Commit f432a7a

Browse files
authored
[NeuralChat]Update the torch version for RAG (intel#1314)
Update the torch version for RAG Signed-off-by: XuhuiRen <[email protected]>
1 parent 525ea86 commit f432a7a

File tree

1 file changed

+2
-0
lines changed
  • intel_extension_for_transformers/neural_chat/pipeline/plugins/retrieval

1 file changed

+2
-0
lines changed

intel_extension_for_transformers/neural_chat/pipeline/plugins/retrieval/requirements.txt

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,4 +14,6 @@ qdrant-client
1414
rank_bm25
1515
scikit-learn
1616
sentence-transformers==2.3.1
17+
torch==2.2.0
18+
torchaudio==2.2.0
1719
unstructured

0 commit comments

Comments
 (0)