requirements.txt 596 B

12345678910111213141516171819202122232425262728293031
  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. asyncpg>=0.29.0
  10. alembic>=1.13.0
  11. python-multipart>=0.0.9
  12. websockets>=12.0
  13. scipy>=1.14.0
  14. scikit-learn>=1.5.0
  15. pillow>=10.4.0
  16. pandas>=2.2.0
  17. pyarrow>=17.0.0
  18. addict>=2.4.0
  19. modelscope>=1.15.0,<1.18.0
  20. oss2>=2.18.0
  21. datasets>=3.0.0
  22. huggingface_hub
  23. aiohttp>=3.9.0,<3.11.0
  24. # GPU 训练依赖
  25. torch>=2.4.0
  26. transformers>=4.45.0
  27. peft>=0.13.0
  28. trl>=0.12.0
  29. accelerate>=1.0.0
  30. bitsandbytes>=0.44.0