requirements.txt 370 B

123456789101112131415161718192021222324
  1. # Core serving
  2. fastapi>=0.104.0
  3. uvicorn[standard]
  4. pydantic>=2.0
  5. # Model inference
  6. torch>=2.1.0
  7. transformers>=4.36.0
  8. # Training
  9. datasets>=2.14.0
  10. accelerate>=0.25.0
  11. peft>=0.7.0
  12. trl>=0.7.0
  13. evaluate>=0.4.0
  14. seqeval>=1.2.5
  15. scikit-learn>=1.3.0
  16. tqdm>=4.66.0
  17. # Chinese tokenizer support
  18. sentencepiece>=0.1.99
  19. # Optional: Qwen LoRA training acceleration
  20. # pip install unsloth