Update requirements.txt

requirements.txt

@@ -1,333 +1,10 @@
-absl-py==2.4.0
+torch
+torchvision
+transformers==4.38.2
 accelerate==0.27.2
-aiobotocore==3.5.0
-aiohappyeyeballs==2.6.1
-aiohttp==3.13.5
-aiohttp-retry==2.9.1
-aioitertools==0.13.0
-aiosignal==1.4.0
-albumentations==1.4.7
-amqp==5.3.1
-annotated-doc==0.0.4
-annotated-types==0.7.0
-antlr4-python3-runtime==4.9.3
-anyio==4.13.0
-appdirs==1.4.4
-archspec @ file:///croot/archspec_1697725767277/work
-argon2-cffi==25.1.0
-argon2-cffi-bindings==25.1.0
-arrow==1.4.0
-asttokens @ file:///opt/conda/conda-bld/asttokens_1646925590279/work
-astunparse==1.6.3
-async-lru==2.3.0
-async-timeout==5.0.1
-asyncssh==2.22.0
-atpublic==7.0.0
-attrs==26.1.0
-babel==2.18.0
-backoff==2.2.1
-beautifulsoup4 @ file:///croot/beautifulsoup4-split_1681493039619/work
-billiard==4.2.4
-bleach==6.3.0
-boltons @ file:///croot/boltons_1677628692245/work
-boto3==1.42.91
-botocore==1.42.91
-Brotli @ file:///tmp/abs_ecyw11_7ze/croots/recipe/brotli-split_1659616059936/work
-celery==5.6.3
-certifi==2026.4.22
-cffi==2.0.0
-chardet @ file:///home/builder/ci_310/chardet_1640804867535/work
-charset-normalizer==3.4.7
-click==8.3.3
-click-didyoumean==0.3.1
-click-plugins==1.1.1.2
-click-repl==0.3.0
-colorama==0.4.6
-comm==0.2.3
-conda @ file:///croot/conda_1696257509808/work
-conda-build @ file:///croot/conda-build_1708025865815/work
-conda-content-trust @ file:///croot/conda-content-trust_1693490622020/work
-conda-libmamba-solver @ file:///croot/conda-libmamba-solver_1691418897561/work/src
-conda-package-handling @ file:///croot/conda-package-handling_1690999929514/work
-conda_index @ file:///croot/conda-index_1706633791028/work
-conda_package_streaming @ file:///croot/conda-package-streaming_1690987966409/work
-configobj==5.0.9
-contourpy==1.3.2
-cryptography @ file:///croot/cryptography_1707523700518/work
-cuda-bindings==12.9.4
-cuda-pathfinder==1.5.4
-cuda-toolkit==12.8.1
-cycler==0.12.1
-dacite==1.6.0
-dagshub==0.7.0
-dagshub-annotation-converter==0.2.0
-dataclasses-json==0.6.7
-datasets==2.18.0
-debugpy==1.8.20
-decorator @ file:///opt/conda/conda-bld/decorator_1643638310831/work
-defusedxml==0.7.1
-dictdiffer==0.9.0
-dill==0.3.8
-diskcache==5.6.3
-distro @ file:///croot/distro_1701455004953/work
-dnspython==2.6.1
-docker-pycreds==0.4.0
-dpath==2.2.0
-dulwich==1.2.0
-dvc==3.67.1
-dvc-data==3.18.3
-dvc-http==2.32.0
-dvc-objects==5.2.0
-dvc-render==1.0.2
-dvc-s3==3.3.0
-dvc-studio-client==0.22.0
-dvc-task==0.40.2
-entrypoints==0.4
-exceptiongroup==1.3.1
-executing @ file:///opt/conda/conda-bld/executing_1646925071911/work
-expecttest==0.2.1
-fastjsonschema==2.21.2
-filelock==3.29.0
-flatten-dict==0.4.2
-flufl.lock==9.0.0
-fonttools==4.62.1
-fqdn==1.5.1
-frozenlist==1.8.0
-fsspec==2024.2.0
-ftfy==6.3.1
-funcy==2.0
-gitdb==4.0.12
-GitPython==3.1.49
-gmpy2 @ file:///tmp/build/80754af9/gmpy2_1645455533097/work
-gql==4.0.0
-grandalf==0.8
-graphql-core==3.2.8
-gto==1.9.0
-h11==0.16.0
-hf-xet==1.4.3
-httpcore==1.0.9
-httpx==0.28.1
-huggingface-hub==0.20.3
-hydra-core==1.3.2
-hypothesis==6.98.10
-idna==3.13
-ImageHash==4.3.1
-ImageIO==2.37.3
-ipykernel==7.2.0
-ipython @ file:///croot/ipython_1704833016303/work
-isoduration==20.11.0
-iterative-telemetry==0.0.10
-jedi @ file:///tmp/build/80754af9/jedi_1644315229345/work
-Jinja2==3.1.6
-jmespath==1.1.0
-joblib==1.5.3
-json5==0.14.0
-jsonpatch @ file:///tmp/build/80754af9/jsonpatch_1615747632069/work
-jsonpointer==2.1
-jsonschema @ file:///croot/jsonschema_1699041609003/work
-jsonschema-specifications @ file:///croot/jsonschema-specifications_1699032386549/work
-jupyter-events==0.12.1
-jupyter-lsp==2.3.1
-jupyter_client==8.8.0
-jupyter_core==5.9.1
-jupyter_server==2.18.2
-jupyter_server_terminals==0.5.4
-jupyterlab==4.5.6
-jupyterlab_pygments==0.3.0
-jupyterlab_server==2.28.0
-kiwisolver==1.5.0
-kombu==5.6.2
-lazy-loader==0.5
-libarchive-c @ file:///tmp/build/80754af9/python-libarchive-c_1617780486945/work
-libmambapy @ file:///croot/mamba-split_1698782620632/work/libmambapy
-lightning-utilities==0.15.3
-lxml==6.1.0
-markdown-it-py==4.0.0
-MarkupSafe==3.0.3
-marshmallow==3.26.2
-matplotlib==3.8.3
-matplotlib-inline @ file:///opt/conda/conda-bld/matplotlib-inline_1662014470464/work
-mdurl==0.1.2
-menuinst @ file:///croot/menuinst_1706732933928/work
-mistune==3.2.1
-mkl-fft @ file:///croot/mkl_fft_1695058164594/work
-mkl-random @ file:///croot/mkl_random_1695059800811/work
-mkl-service==2.4.0
-more-itertools @ file:///croot/more-itertools_1700662129964/work
-mpmath==1.3.0
-multidict==6.7.1
-multiprocess==0.70.16
-mypy_extensions==1.1.0
-nbclient==0.10.4
-nbconvert==7.17.1
-nbformat==5.10.4
-nest-asyncio==1.6.0
-networkx==3.3
-nltk==3.8.1
-notebook_shim==0.2.4
-numpy==1.26.4
-nvidia-cublas==13.1.0.3
-nvidia-cublas-cu12==12.8.4.1
-nvidia-cuda-cupti==13.0.85
-nvidia-cuda-cupti-cu12==12.8.90
-nvidia-cuda-nvrtc==13.0.88
-nvidia-cuda-nvrtc-cu12==12.8.93
-nvidia-cuda-runtime==13.0.96
-nvidia-cuda-runtime-cu12==12.8.90
-nvidia-cudnn-cu12==9.19.0.56
-nvidia-cudnn-cu13==9.19.0.56
-nvidia-cufft==12.0.0.61
-nvidia-cufft-cu12==11.3.3.83
-nvidia-cufile==1.15.1.6
-nvidia-cufile-cu12==1.13.1.3
-nvidia-curand==10.4.0.35
-nvidia-curand-cu12==10.3.9.90
-nvidia-cusolver==12.0.4.66
-nvidia-cusolver-cu12==11.7.3.90
-nvidia-cusparse==12.6.3.3
-nvidia-cusparse-cu12==12.5.8.93
-nvidia-cusparselt-cu12==0.7.1
-nvidia-cusparselt-cu13==0.8.0
-nvidia-nccl-cu12==2.28.9
-nvidia-nccl-cu13==2.28.9
-nvidia-nvjitlink==13.0.88
-nvidia-nvjitlink-cu12==12.8.93
-nvidia-nvshmem-cu12==3.4.5
-nvidia-nvshmem-cu13==3.4.5
-nvidia-nvtx==13.0.85
-nvidia-nvtx-cu12==12.8.90
-omegaconf==2.3.0
-open-clip-torch==2.24.0
-opencv-python-headless
-optree==0.10.0
-orjson==3.11.8
-overrides==7.7.0
-packaging==26.2
-pandas==2.1.4
-pandocfilters==1.5.1
-parso @ file:///opt/conda/conda-bld/parso_1641458642106/work
-pathspec==1.1.1
-pathvalidate==3.3.1
-pexpect @ file:///tmp/build/80754af9/pexpect_1605563209008/work
 pillow==10.3.0
-pkginfo @ file:///croot/pkginfo_1679431160147/work
-platformdirs @ file:///croot/platformdirs_1692205439124/work
-pluggy @ file:///tmp/build/80754af9/pluggy_1648024709248/work
-portalocker==3.2.0
-prometheus_client==0.25.0
-prompt-toolkit @ file:///croot/prompt-toolkit_1704404351921/work
-propcache==0.4.1
-protobuf==4.25.9
-psutil==7.2.2
-ptyprocess @ file:///tmp/build/80754af9/ptyprocess_1609355006118/work/dist/ptyprocess-0.7.0-py2.py3-none-any.whl
-pure-eval @ file:///opt/conda/conda-bld/pure_eval_1646925070566/work
-pyarrow==24.0.0
-pyarrow-hotfix==0.7
-pycocoevalcap==1.2
-pycocotools==2.0.11
-pycosat @ file:///croot/pycosat_1696536503704/work
-pycparser @ file:///tmp/build/80754af9/pycparser_1636541352034/work
-pydantic==2.9.2
-pydantic-settings==2.14.0
-pydantic_core==2.23.4
-pydot==4.0.1
-pygit2==1.18.2
-Pygments @ file:///croot/pygments_1684279966437/work
-pygtrie==2.5.0
-pyOpenSSL @ file:///croot/pyopenssl_1708380408460/work
-pyparsing==3.3.2
-PySocks @ file:///home/builder/ci_310/pysocks_1640793678128/work
-python-dateutil==2.9.0.post0
-python-dotenv==1.2.2
-python-etcd==0.4.5
-python-json-logger==4.1.0
-pytz==2026.1.post1
-PyWavelets==1.8.0
-PyYAML==6.0.1
-pyzmq==27.1.0
-referencing @ file:///croot/referencing_1699012038513/work
-regex==2026.4.4
-requests==2.32.3
-requests-toolbelt==1.0.0
-rfc3339-validator==0.1.4
-rfc3986-validator==0.1.1
-rich==15.0.0
-rouge_score==0.1.2
-rpds-py @ file:///croot/rpds-py_1698945930462/work
-ruamel.yaml @ file:///croot/ruamel.yaml_1666304550667/work
-ruamel.yaml.clib @ file:///croot/ruamel.yaml.clib_1666302247304/work
-s3fs==2026.3.0
-s3transfer==0.16.1
-sacrebleu==2.4.0
+numpy==1.26.4
+timm==0.9.12
 safetensors==0.4.2
-scikit-image==0.25.2
-scikit-learn==1.7.2
-scipy==1.15.3
-scmrepo==3.6.2
-semver==3.0.4
-Send2Trash==2.1.0
-sentence-transformers==2.7.0
 sentencepiece==0.2.0
-sentry-sdk==2.58.0
-setproctitle==1.3.7
-shellingham==1.5.4
-shortuuid==1.0.13
-shtab==1.8.0
-six==1.17.0
-smmap==5.0.3
-sortedcontainers==2.4.0
-soupsieve @ file:///croot/soupsieve_1696347547217/work
-sqltrie==0.11.2
-stack-data @ file:///opt/conda/conda-bld/stack_data_1646927590127/work
-sympy==1.14.0
-tabulate==0.10.0
-tenacity==9.1.4
-terminado==0.18.1
-threadpoolctl==3.6.0
-tifffile==2025.5.10
-timm==0.9.12
-tinycss2==1.4.0
-tokenizers==0.15.2
-tomli @ file:///opt/conda/conda-bld/tomli_1657175507142/work
-tomlkit==0.14.0
-toolz @ file:///croot/toolz_1667464077321/work
-torch
-torchaudio
-torchelastic==0.2.2
-torchmetrics==1.9.0
-torchvision
-tornado==6.5.5
-tqdm==4.66.2
-traitlets @ file:///croot/traitlets_1671143879854/work
-transformers==4.38.2
-treelib==1.8.0
-triton==3.6.0
-truststore @ file:///croot/truststore_1695244293384/work
-typer==0.25.0
-types-dataclasses==0.6.6
-typing-inspect==0.9.0
-typing-inspection==0.4.2
-typing_extensions==4.10.0
-tzdata==2026.2
-tzlocal==5.3.1
-uri-template==1.3.0
-urllib3==2.6.3
-vine==5.1.0
-voluptuous==0.16.0
-wandb==0.26.1
-wcwidth==0.7.0
-webcolors==25.10.0
-webencodings==0.5.1
-websocket-client==1.9.0
-wrapt==2.1.2
-xxhash==3.7.0
-yarl==1.23.0
-zc.lockfile==4.0
-zstandard @ file:///croot/zstandard_1677013143055/work
-notebook==7.5.0
-einops==0.8.2
-umap-learn==0.5.12
-grad-cam==1.5.5
-ttach==0.0.3
-fastapi==0.112.4
-starlette==0.38.6
+opencv-python-headless