--extra-index-url https://download.pytorch.org/whl/cu121
--extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/ # https://github.com/microsoft/onnxruntime/issues/21684

# add aliyun mirror
# --index-url https://mirrors.aliyun.com/pypi/simple/
# pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu121 

rotary_embedding_torch==0.8.6
timm

torch==2.5.1
torchvision==0.20.1
torchaudio==2.5.1

# numpy==1.24.0
# byted-wandb
# mpi4py

transformers==4.47
datasets
huggingface_hub[cli]
accelerate
einops
wandb
dask[dataframe]
wandb
dask[dataframe]
rouge
omegaconf
tyro
joblib
tyro
joblib
python-box
pycocotools
jsonargparse[signatures]
tensorboard
einops
# pytorch-lightning

pyarrow
peft
deepspeed
av
decord
opencv-python
opencv-python-headless
open_clip_torch
# s2-wrapper
# git+https://ghproxy.cn/https://github.com/descriptinc/audiotools
pytorch_lightning
librosa

cryptography

# Glm
conformer
hydra-core
diffusers
lightning
gdown
wget
pandas
hyperpyyaml

# Renderer
mediapipe==0.10.11
imgaug==0.4.0
numpy==1.26.4
hyperpyyaml

# Cosyvoice
# onnxruntime-gpu==1.18.0; sys_platform == 'linux'
onnxruntime-gpu; sys_platform == 'linux'
# onnxruntime==1.18.0; sys_platform == 'darwin' or sys_platform == 'windows'
openai-whisper
# openai-whisper==20231117
WeTextProcessing==1.0.3
modelscope
# modelscope==1.15.0

