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