requirements.txt 580 B

123456789101112131415161718192021222324252627282930
  1. # --- 3090 单机部署依赖(前后端合一,含 GPU 训练栈) ---
  2. fastapi>=0.115.0
  3. uvicorn[standard]>=0.30.0
  4. pydantic>=2.0
  5. pydantic-settings>=2.0
  6. python-dotenv>=1.0
  7. sqlalchemy[asyncio]>=2.0
  8. aiosqlite>=0.20.0
  9. alembic>=1.13.0
  10. python-multipart>=0.0.9
  11. websockets>=12.0
  12. scipy>=1.14.0
  13. scikit-learn>=1.5.0
  14. pillow>=10.4.0
  15. pandas>=2.2.0
  16. pyarrow>=17.0.0
  17. addict>=2.4.0
  18. modelscope>=1.15.0,<1.18.0
  19. oss2>=2.18.0
  20. datasets>=3.0.0
  21. huggingface_hub
  22. aiohttp>=3.9.0,<3.11.0
  23. # GPU 训练依赖
  24. torch>=2.4.0
  25. transformers>=4.45.0
  26. peft>=0.13.0
  27. trl>=0.12.0
  28. accelerate>=1.0.0
  29. bitsandbytes>=0.44.0