torch==2.0.1
torchvision==0.15.2
torchaudio==2.0.2
transformers==4.31.0
sentencepiece==0.1.99
shortuuid
accelerate==0.21.0
peft==0.4.0
bitsandbytes==0.41.0
pydantic
markdown2[all]
numpy
scikit-learn==1.2.2
gradio==4.16.0
gradio_client==0.8.1
requests
httpx==0.24.0
uvicorn
fastapi
einops==0.6.1
einops-exts==0.0.4
timm==0.6.13
deepspeed==0.12.6
ninja
wandb
build
twine
pytorchvideo==0.1.5
ftfy
regex
fvcore
decord==0.6.0
iopath
matplotlib
types-regex
vtk
mayavi
cartopy
moviepy
omegaconf
librosa
opencv-python
easydict
# pip install flash-attn==2.3.5 --no-build-isolation
# pip install /yeesuanAI05/thumt/fbh/flash_attn-2.3.5+cu117torch2.0cxx11abiFALSE-cp310-cp310-linux_x86_64.whl