24 lines
522 B
Plaintext
24 lines
522 B
Plaintext
# 基础依赖
|
||
faiss-cpu==1.11.0
|
||
fastapi
|
||
uvicorn
|
||
gunicorn
|
||
pydantic
|
||
numpy==1.24.4
|
||
|
||
# 深度学习框架(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
|
||
llama-index-readers-file
|