Fine-tuning/backend/requirements.txt

fastapi>=0.115.0
uvicorn[standard]>=0.30.0
pydantic>=2.0
pydantic-settings>=2.0
python-dotenv>=1.0
sqlalchemy[asyncio]>=2.0
aiosqlite>=0.20.0
alembic>=1.13.0
python-multipart>=0.0.9
websockets>=12.0
# GPU 相关依赖需要根据实际显卡手动安装：
#   NVIDIA GPU: pip install torch --index-url https://download.pytorch.org/whl/cu121
#   沐曦 GPU:    pip install torch_musa --index-url https://release.mthreads.com/repo/pypi/simple
# 以下为框架和工具库依赖
# transformers>=4.45.0
transformers @ git+https://github.com/huggingface/transformers.git
peft>=0.13.0
trl>=0.12.0
datasets>=3.0.0
# accelerate>=1.0.0          # 随 GPU 环境一起安装
# bitsandbytes>=0.44.0        # 仅 NVIDIA GPU 需要，沐曦环境可能不兼容
scipy>=1.14.0
scikit-learn>=1.5.0
pillow>=10.4.0
huggingface_hub>=0.25.0
modelscope>=1.15.0
addict>=2.4.0
pandas>=2.2.0
pyarrow>=17.0.0
sentencepiece>=0.2.0
protobuf>=4.25.0