requirements.txt 843 B

1234567891011121314151617181920212223242526272829303132333435363738
  1. # 核心深度学习框架(必须按顺序安装)
  2. torch>=2.0.0
  3. transformers>=4.30.0
  4. datasets>=2.12.0
  5. peft>=0.4.0
  6. bitsandbytes>=0.39.0
  7. accelerate>=0.20.0
  8. # Hugging Face生态系统(必需)
  9. huggingface_hub>=0.16.0
  10. safetensors>=0.3.0
  11. # 分词器和模型支持(必需)
  12. sentencepiece>=0.1.99
  13. tokenizers>=0.13.0
  14. protobuf>=3.20.0
  15. # flash-attn>=2.0.0 # 可选:需要CUDA和较长编译时间,如需要请手动安装
  16. # 数据处理和网络请求(必需)
  17. requests>=2.31.0
  18. beautifulsoup4>=4.11.0
  19. numpy>=1.21.0
  20. pandas>=1.5.0
  21. tqdm>=4.65.0
  22. jieba>=0.42.1
  23. # 模型源支持(必需)
  24. modelscope>=1.9.0
  25. # 系统监控和环境检查(必需)
  26. psutil>=5.9.0
  27. nvidia-ml-py>=12.535.108
  28. # 可选依赖(实验跟踪和可视化,可根据需要安装)
  29. # wandb>=0.15.0
  30. # tensorboard>=2.10.0
  31. # matplotlib>=3.5.0
  32. # seaborn>=0.11.0