alphapen_trocr_large_30000 / requirements.txt
hadrakey's picture
Training in progress, step 1000
5486f6f verified
accelerate==0.30.1
aiohttp==3.9.5
aiosignal==1.3.1
attrs==23.2.0
bitsandbytes==0.43.1
Brotli @ file:///croot/brotli-split_1714483155106/work
certifi @ file:///croot/certifi_1717618050233/work/certifi
chardet==5.2.0
charset-normalizer @ file:///tmp/build/80754af9/charset-normalizer_1630003229654/work
datasets==2.19.2
dill==0.3.8
docstring_parser==0.16
einops==0.8.0
evaluate==0.4.2
filelock @ file:///croot/filelock_1700591183607/work
flash-attn==2.5.8
frozenlist==1.4.1
fsspec==2024.3.1
gmpy2 @ file:///work/ci_py311/gmpy2_1676839849213/work
huggingface-hub==0.23.3
idna @ file:///croot/idna_1714398848350/work
Jinja2 @ file:///croot/jinja2_1716993405101/work
joblib==1.4.2
lightning-utilities==0.11.2
markdown-it-py==3.0.0
MarkupSafe @ file:///croot/markupsafe_1704205993651/work
mdurl==0.1.2
mkl-fft @ file:///croot/mkl_fft_1695058164594/work
mkl-random @ file:///croot/mkl_random_1695059800811/work
mkl-service==2.4.0
mpmath @ file:///croot/mpmath_1690848262763/work
multidict==6.0.5
multiprocess==0.70.16
networkx @ file:///croot/networkx_1717597493534/work
ninja==1.11.1.1
numpy @ file:///croot/numpy_and_numpy_base_1708638617955/work/dist/numpy-1.26.4-cp311-cp311-linux_x86_64.whl#sha256=5f96f274d410a1682519282ae769c877d32fdbf171aa8badec7bf5e1d3a1748a
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==2.20.5
nvidia-nvjitlink-cu12==12.5.40
nvidia-nvtx-cu12==12.1.105
packaging==24.0
pandas==2.2.2
peft==0.11.1
pillow @ file:///croot/pillow_1714398848491/work
psutil==5.9.8
pyarrow==16.1.0
pyarrow-hotfix==0.6
Pygments==2.18.0
PySocks @ file:///work/ci_py311/pysocks_1676822712504/work
python-dateutil==2.9.0.post0
pytz==2024.1
PyYAML @ file:///croot/pyyaml_1698096049011/work
regex==2024.5.15
requests @ file:///croot/requests_1716902831423/work
rich==13.7.1
safetensors==0.4.3
scikit-learn==1.5.0
scipy==1.13.1
shtab==1.7.1
six==1.16.0
sympy==1.12.1
threadpoolctl==3.5.0
tokenizers==0.19.1
torch==2.3.1
torchaudio==2.3.1
torchmetrics==1.4.0.post0
torchvision==0.18.1
tqdm==4.66.4
transformers @ git+https://github.com/huggingface/transformers.git@60bb571e993b7d73257fb64044726b569fef9403
triton==2.3.1
trl @ git+https://github.com/huggingface/trl@275d33b3ef4f7afd40f79cc53591659bacfa3499
typing_extensions==4.12.1
tyro==0.8.4
tzdata==2024.1
urllib3 @ file:///croot/urllib3_1715635851070/work
xxhash==3.4.1
yarl==1.9.4