requirements.txt

absl-py==2.3.1
accelerate==0.27.2
addict==2.4.0
annotated-types==0.7.0
antlr4-python3-runtime==4.9.3
black==25.1.0
certifi==2022.12.7
charset-normalizer==2.1.1
click==8.2.1
cloudpickle==3.1.1
conda-pack @ file:///home/conda/feedstock_root/build_artifacts/conda-pack_1734292804024/work
contourpy==1.3.2
cycler==0.12.1
deepspeed==0.10.0
einops==0.8.1
filelock==3.13.1
fonttools==4.59.1
fsspec==2024.6.1
fvcore==0.1.5.post20221221
gitdb==4.0.12
GitPython==3.1.45
grpcio==1.74.0
h11==0.16.0
hf-xet==1.1.8
hjson==3.1.0
huggingface-hub==0.34.4
hydra-core==1.3.2
idna==3.4
imageio==2.37.0
iopath==0.1.9
Jinja2==3.1.4
joblib==1.5.1
kiwisolver==1.4.9
lazy_loader==0.4
Markdown==3.8.2
MarkupSafe==2.1.5
matplotlib==3.10.5
mpmath==1.3.0
mypy_extensions==1.1.0
networkx==3.3
ninja==1.13.0
numpy==1.26.4
nvidia-ml-py==12.575.51
omegaconf==2.3.0
opencv-python==4.12.0.88
packaging==25.0
pandas==2.3.2
pathspec==0.12.1
peft==0.5.0
pillow==11.0.0
platformdirs==4.4.0
portalocker==3.2.0
protobuf==6.32.0
psutil==7.0.0
py-cpuinfo==9.0.0
pycocotools==2.0.10
pydantic==1.10.22
pydantic_core==2.33.2
pynvml==12.0.0
pyparsing==3.2.3
python-dateutil==2.9.0.post0
pytz==2025.2
PyYAML==6.0.2
regex==2025.7.34
requests==2.28.1
safetensors==0.6.2
scikit-image==0.25.2
scikit-learn==1.2.2
scipy==1.15.3
seaborn==0.13.2
sentencepiece==0.1.99
sentry-sdk==2.35.1
shapely==2.1.1
six==1.17.0
smmap==5.0.2
sympy==1.13.3
tabulate==0.9.0
tensorboard==2.20.0
tensorboard-data-server==0.7.2
termcolor==3.1.0
threadpoolctl==3.6.0
tifffile==2025.5.10
timm==0.6.13
tokenizers==0.15.2
tomli==2.2.1
torch==2.1.0+cu121
torchaudio==2.1.0+cu121
torchvision==0.16.0+cu121
tqdm==4.67.1
transformers==4.38.2
triton==2.1.0
typing-inspection==0.4.1
typing_extensions==4.12.2
tzdata==2025.2
urllib3==1.26.13
uvicorn==0.35.0
wandb==0.21.1
Werkzeug==3.1.3
wordcloud==1.9.4
yacs==0.1.8