Query-decompose-baselines / requirements.txt
Veblen34's picture
Upload folder using huggingface_hub
cd6775d verified
Raw
History Blame Contribute Delete
965 Bytes
# ===== 查询分割 baseline 依赖(进程内函数版)=====
# ①②③ 直接 import 调用,无需启动任何本地服务。
# 通用
requests
# ① 监督式(本地 Llama+LoRA,transformers+peft 进程内加载)
torch # 按你的 CUDA 版本安装对应 wheel
transformers>=4.40
peft # LoRA 适配器加载
accelerate # device_map="auto" 需要
# ② 无监督(本地 XLM seq2seq),需 GPU;服务代码依赖:
# (XLM 相关模块已内置在 methods/unsupervised/XLM/src,无需 pip)
# ③ ICL(Socratic),默认调外部 gpt-4o-mini:
openai>=1.0 # 也用于 ④ deepseek / 可选的本地 vLLM OpenAI 接口
flask # 仅当你想以“服务版”运行(start/*.sh)时才需要
# ④ SearChain(服务版,进阶):flask + openai + ColBERT 检索栈(colbert-ai、faiss 等)+ 索引
python-dotenv # ColBERT 服务 load_dotenv