.

requirements.txt

@@ -1,13 +1,22 @@
-faiss-cpu
+# 基础依赖
+faiss-cpu==1.11.0
 fastapi
 uvicorn
 gunicorn
 pydantic
-numpy
+numpy==1.24.4
-transformers
-torch
 
-# 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