requirements.txt

absl-py==2.1.0
accelerate
aiofiles==23.2.1
aiohttp==3.9.5
aiosignal==1.3.1
alembic==1.13.2
altair==5.4.0
annotated-types==0.7.0
antlr4-python3-runtime==4.9.3
anyio==4.4.0

astunparse==1.6.3
attrs==23.2.0
banal==1.0.6
bleach==6.1.0
blinker==1.8.2
blis==0.7.11
braceexpand==0.1.7
cachetools==5.5.0
catalogue==2.0.10
certifi==2024.6.2
cfgv==3.4.0
charset-normalizer==3.3.2
click==8.1.7
cloudpathlib==0.18.1

confection==0.1.5
contexttimer==0.3.3
contourpy==1.2.1
cycler==0.12.1
cymem==2.0.8
datasets==2.20.0

decord==0.6.0
diffusers==0.31.0
transformers==4.46.3

dill==0.3.8
distlib==0.3.8
distro==1.9.0
docker-pycreds==0.4.0
einops==0.8.0

fairscale
fastapi==0.112.0
ffmpy==0.4.0
filelock==3.15.1
Flask==3.0.3
Flask-Cors==4.0.1
flatbuffers==24.3.25
fonttools==4.53.0
frozenlist==1.4.1
fsspec==2024.5.0
ftfy==6.2.3
gast==0.6.0
gitdb==4.0.11
gitlab==1.0.2
GitPython==3.1.43
google-pasta==0.2.0
gradio==4.41.0
gradio_client==1.3.0
greenlet==3.0.3
grpcio==1.64.1
h11==0.14.0
h5py==3.11.0
httpcore==1.0.5
httpx==0.27.0
huggingface-hub==0.23.4
identify==2.6.0
idna==3.7
imageio==2.34.1
importlib_metadata==7.1.0
importlib_resources==6.4.0
invisible-watermark
iopath==0.1.10

itsdangerous==2.2.0

Jinja2==3.1.4
joblib==1.4.2
jsonlines==4.0.0
jsonschema==4.23.0
jsonschema-specifications==2023.12.1

kaggle==1.6.17
keras==3.4.1
kiwisolver==1.4.5
langcodes==3.4.0
language_data==1.2.0
lazy_loader==0.4
libclang==18.1.1
lightning-utilities==0.11.6
lmdb==1.4.1
lmdbdict==0.2.2
Mako==1.3.5
marisa-trie==1.2.0
Markdown==3.6
markdown-it-py==3.0.0
MarkupSafe==2.1.5
matplotlib==3.9.0

mdurl==0.1.2
ml-dtypes==0.4.0
modelscope==1.16.1
mpmath==1.3.0
multidict==6.0.5
multiprocess==0.70.16
murmurhash==1.0.10
namex==0.0.8
narwhals==1.4.2

networkx
nodeenv==1.9.1
numpy==1.26.4
nvidia-cublas-cu12==12.1.3.1
nvidia-cuda-cupti-cu12==12.1.105
nvidia-cuda-nvrtc-cu12==12.1.105
nvidia-cuda-runtime-cu12==12.1.105
nvidia-cudnn-cu12==8.9.2.26
nvidia-cufft-cu12==11.0.2.54
nvidia-curand-cu12==10.3.2.106
nvidia-cusolver-cu12==11.4.5.107
nvidia-cusparse-cu12==12.1.0.106
nvidia-nccl-cu12
nvidia-nvjitlink-cu12==12.5.40
nvidia-nvtx-cu12==12.1.105
omegaconf==2.3.0
openai==0.28.0
opencv-python-headless==4.5.5.64
opendatasets==0.1.22
opt-einsum==3.3.0
optree==0.12.1
orjson==3.10.7

pandas==2.2.2

peft

pillow==10.3.0
piq==0.8.0

plotly==5.23.0
portalocker==2.10.1
pre-commit==3.8.0
preshed==3.0.9

protobuf==4.25.3
psutil==6.0.0

pyarrow==17.0.0
pyarrow-hotfix==0.6
pycocoevalcap==1.2
pycocotools==2.0.8
pydantic==2.8.2
pydantic_core==2.20.1
pydeck==0.9.1
pydub==0.25.1

pyparsing==3.1.2

python-gitlab==4.6.0
python-magic==0.4.27
python-multipart==0.0.9
python-slugify==8.0.4
pytz==2024.1
PyYAML==6.0.1

referencing==0.35.1
regex==2024.5.15
requests==2.32.3
requests-toolbelt==1.0.0
rich==13.7.1
rpds-py==0.20.0
ruff==0.5.7
safetensors==0.4.3
salesforce-lavis
scikit-image==0.24.0
scikit-learn==1.5.0
scipy==1.13.1
seaborn==0.13.2
semantic-version==2.10.0
sentencepiece==0.2.0
sentry-sdk==2.10.0
setproctitle==1.3.3
shellingham==1.5.4

smart-open==7.0.4
smmap==5.0.1
sniffio==1.3.1
spacy==3.7.5
spacy-legacy==3.0.12
spacy-loggers==1.0.5
SQLAlchemy==1.4.52
srsly==2.4.8

starlette==0.37.2
streamlit==1.37.1
sympy==1.12.1
tenacity==8.5.0
tensorboard==2.17.0
tensorboard-data-server==0.7.2
tensorflow==2.17.0
tensorflow-io-gcs-filesystem==0.37.1
termcolor==2.4.0
text-unidecode==1.3
thinc==8.2.5
threadpoolctl==3.5.0
tifffile==2024.6.18
timm
tokenizers
toml==0.10.2
tomlkit==0.12.0
torch==2.2.0
torchmetrics
torchstat
torchsummary==1.5.1
torchvision

tqdm==4.66.4


triton==2.2.0
typer==0.12.3

tzdata==2024.1
urllib3==2.2.2
uvicorn==0.30.5
virtualenv==20.26.3
wandb==0.17.5
wasabi==1.1.3
watchdog==4.0.2

weasel==0.4.1
webdataset==0.2.96
webencodings==0.5.1
websockets==12.0
Werkzeug==3.0.3
wrapt==1.16.0
xformer==1.0.1
xxhash==3.4.1
yarl==1.9.4