From a00a6f2d2c54f7db7a2c628eb01ba801a4d6793c Mon Sep 17 00:00:00 2001
From: XuhuiRen <44249229+XuhuiRen@users.noreply.github.com>
Date: Tue, 9 Apr 2024 09:44:48 +0800
Subject: [PATCH] [NeuralChat] Add restriction for torch version  (#1458)

---
 .../neural_chat/pipeline/plugins/retrieval/requirements.txt     | 2 ++
 1 file changed, 2 insertions(+)

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
index ab8c6ac032a..2a5af3ef179 100644
--- 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]