Spaces:
Running
Running
File size: 693 Bytes
c1c9e88 7066d20 c1c9e88 7066d20 c1c9e88 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 |
# 核心深度学习框架 torch>=2.0.0 torchvision xformers # Transformers生态 transformers>=4.35.0 accelerate>=0.24.0 tokenizers huggingface_hub # 数据处理 datasets>=2.14.0 numpy>=1.24.0 sentence-transformers faiss-cpu # 模型微调和RLHF peft>=0.9.0 trl[peft]>=0.7.10 bitsandbytes>=0.41.0 # 图像生成 diffusers invisible_watermark # 数据标注 label-studio # API和网络请求 flickrapi requests # 实验跟踪和可视化 wandb>=0.15.0 tensorboard>=2.13.0 # 评估指标 evaluate sacrebleu rouge-score # 系统工具和监控 scipy protobuf sentencepiece alive_progress psutil gpustat # 高级优化器（可选） deepspeed>=0.10.0 # RLHF特定工具 reward-bench |