[NeuralChat] Add restriction for torch version (#1458)

intel · Apr 9, 2024 · a00a6f2 · a00a6f2
1 parent 1c65507
commit a00a6f2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/intel_extension_for_transformers/neural_chat/pipeline/plugins/retrieval/requirements.txt b/intel_extension_for_transformers/neural_chat/pipeline/plugins/retrieval/requirements.txt
@@ -14,4 +14,6 @@ qdrant-client
 rank_bm25
 scikit-learn
 sentence-transformers==2.3.1
+torch==2.2.0
+torchaudio==2.2.0
 unstructured[all-docs]