From dc5428da62a42eb80c62fb864071cbd03b58b800 Mon Sep 17 00:00:00 2001
From: hailin <hailin@gdzx.xyz>
Date: Thu, 8 May 2025 17:25:13 +0800
Subject: [PATCH] .

---
 requirements.txt | 19 ++++++++++++++-----
 1 file changed, 14 insertions(+), 5 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index 50a2a68..c1cb1e9 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,13 +1,22 @@
-faiss-cpu
+# 基础依赖
+faiss-cpu==1.11.0
 fastapi
 uvicorn
 gunicorn
 pydantic
-numpy
-transformers
-torch
+numpy==1.24.4
 
-# llama-index 核心 + 子模块
+# 深度学习框架（CUDA 12.1 版本，适配 CUDA 12.4 系统）
+torch==2.1.0+cu121
+torchvision==0.16.0+cu121
+torchaudio==2.1.0+cu121
+--extra-index-url https://download.pytorch.org/whl/cu121
+
+# Transformers 与向量模型
+transformers
+sentence-transformers
+
+# LlamaIndex 核心 + 模块化子包（0.12.34 分裂架构）
 llama-index==0.12.34
 llama-index-embeddings-huggingface
 llama-index-vector-stores-faiss