requirements.txt 294 B

123456789101112131415161718192021
  1. # EffectiveRank 项目依赖
  2. # ========================
  3. # 核心计算
  4. torch>=2.0.0
  5. numpy>=1.24.0
  6. # 数据处理
  7. datasets>=2.14.0
  8. huggingface_hub>=0.20.0
  9. # 可视化
  10. matplotlib>=3.7.0
  11. seaborn>=0.12.0
  12. # 工具
  13. typing-extensions>=4.0.0
  14. # 模型推理
  15. transformers>=4.40.0
  16. accelerate>=0.20.0