Requirements for test UNET

requirements.txt

@@ -0,0 +1,163 @@
+accelerate==1.1.1
+aiofiles==23.2.1
+aiohttp==3.9.1
+aiosignal==1.3.1
+altair==5.3.0
+annotated-types==0.6.0
+astunparse==1.6.3
+async-timeout==4.0.3
+attrs==23.2.0
+beautifulsoup4==4.12.3
+braceexpand==0.1.7
+Brotli @ file:///home/conda/feedstock_root/build_artifacts/brotli-split_1725267488082/work
+certifi @ file:///home/conda/feedstock_root/build_artifacts/certifi_1734380492396/work/certifi
+cffi @ file:///home/conda/feedstock_root/build_artifacts/cffi_1725560520483/work
+charset-normalizer @ file:///home/conda/feedstock_root/build_artifacts/charset-normalizer_1733218092148/work
+click==8.1.7
+colored==2.2.4
+coloredlogs==15.0.1
+contourpy==1.2.1
+cuda-python==12.1.0
+cycler==0.12.1
+Cython==3.0.11
+diffusers==0.30.0
+docker-pycreds==0.4.0
+easydict==1.9
+einops==0.7.0
+exceptiongroup==1.2.1
+ffmpy==0.3.2
+filelock==3.13.1
+flatbuffers==24.3.25
+fonttools==4.51.0
+frozenlist==1.4.1
+fsspec==2024.3.1
+future==1.0.0
+gdown==5.2.0
+gitdb==4.0.11
+GitPython==3.1.43
+gmpy2 @ file:///home/conda/feedstock_root/build_artifacts/gmpy2_1733462536562/work
+google-pasta==0.2.0
+h2 @ file:///home/conda/feedstock_root/build_artifacts/h2_1733298745555/work
+h5py==3.11.0
+hpack @ file:///home/conda/feedstock_root/build_artifacts/hpack_1733299205993/work
+huggingface-hub==0.26.3
+humanfriendly==10.0
+hyperframe @ file:///home/conda/feedstock_root/build_artifacts/hyperframe_1733298771451/work
+idna @ file:///home/conda/feedstock_root/build_artifacts/idna_1733211830134/work
+imageio==2.34.0
+imgaug==0.4.0
+importlib_metadata==7.1.0
+importlib_resources==6.4.0
+Jinja2==3.1.2
+joblib==1.3.2
+jsonschema-specifications==2023.12.1
+kiwisolver==1.4.5
+lazy_loader==0.4
+libclang==18.1.1
+lightning==2.4.0
+lightning-utilities==0.11.8
+Markdown==3.6
+markdown-it-py==3.0.0
+MarkupSafe==2.1.3
+matplotlib==3.8.4
+mdurl==0.1.2
+mpmath==1.3.0
+multidict==6.0.4
+multiprocess==0.70.15
+namex==0.0.8
+networkx==3.2.1
+ninja==1.11.1.2
+numpy @ file:///home/conda/feedstock_root/build_artifacts/numpy_1733688207250/work/dist/numpy-2.2.0-cp310-cp310-linux_x86_64.whl#sha256=093fcb0a49217a9cc2cd41219a4055c86e260aca12d8cd27b58bae10ad3a350b
+nvidia-ammo==0.7.0
+nvidia-cublas-cu11==11.10.3.66
+nvidia-cuda-cupti-cu11==11.7.101
+nvidia-cuda-nvrtc-cu11==11.7.99
+nvidia-cuda-runtime-cu11==11.7.99
+nvidia-cudnn-cu11==8.5.0.96
+nvidia-cufft-cu11==10.9.0.58
+nvidia-curand-cu11==10.2.10.91
+nvidia-cusolver-cu11==11.4.0.1
+nvidia-cusparse-cu11==11.7.4.91
+nvidia-nccl-cu11==2.14.3
+nvidia-nvjitlink-cu12==12.4.127
+nvidia-nvtx-cu11==11.7.91
+onnx==1.17.0
+onnx-graphsurgeon==0.5.2
+onnxruntime-gpu==1.20.1
+onnxsim==0.4.36
+opt-einsum==3.3.0
+optree==0.11.0
+orjson==3.10.1
+packaging==24.2
+pillow @ file:///home/conda/feedstock_root/build_artifacts/pillow_1729065631911/work
+platformdirs==4.3.6
+polygraphy==0.49.9
+protobuf==5.29.0
+psutil==6.1.0
+pyarrow-hotfix==0.6
+pycparser @ file:///home/conda/feedstock_root/build_artifacts/bld/rattler-build_pycparser_1733195786/work
+pydub==0.25.1
+Pygments==2.17.2
+pyparsing==3.1.2
+PySocks @ file:///home/conda/feedstock_root/build_artifacts/pysocks_1733217236728/work
+python-dateutil==2.8.2
+pytorch-lightning==2.4.0
+PyYAML==6.0.1
+rawpy==0.19.1
+referencing==0.34.0
+regex==2023.12.25
+requests @ file:///home/conda/feedstock_root/build_artifacts/requests_1733217035951/work
+responses==0.18.0
+rich==13.7.1
+rouge-score==0.1.2
+rpds-py==0.18.0
+ruff==0.4.1
+safetensors==0.4.5
+scipy==1.14.1
+seaborn==0.13.2
+-e git+https://github.com/geekyutao/Inpaint-Anything.git@5bfa9f3d1a829d71bddecf28c228952dbf825e3a#egg=segment_anything&subdirectory=segment_anything
+semantic-version==2.10.0
+sentry-sdk==2.17.0
+setproctitle==1.3.3
+shapely==2.0.4
+shellingham==1.5.4
+six==1.16.0
+smmap==5.0.1
+sniffio==1.3.1
+soupsieve==2.5
+sympy==1.13.1
+tabulate==0.9.0
+tensorboard-data-server==0.7.2
+tensorboardX==2.6.2.2
+tensorrt==8.6.1
+tensorrt-bindings==8.6.1
+tensorrt-libs==8.6.1
+termcolor==2.4.0
+threadpoolctl==3.4.0
+tifffile==2024.4.18
+tokenizers==0.21.0
+tomlkit==0.12.0
+torch==2.5.1
+torchaudio==2.5.1
+torchmetrics==1.6.0
+torchprofile==0.0.4
+torchvision==0.20.1
+tqdm==4.67.1
+transformers==4.47.1
+transnetv2 @ git+https://github.com/soCzech/TransNetV2.git@85cef72af9a916bdfd7cc94a670c9cdfbf12d1ed
+triton==3.1.0
+typer==0.12.3
+typing_extensions==4.12.2
+tzdata==2023.4
+urllib3 @ file:///home/conda/feedstock_root/build_artifacts/urllib3_1733206856906/work
+uvicorn==0.29.0
+wandb==0.18.5
+webdataset==0.2.86
+websockets==11.0.3
+Werkzeug==3.0.2
+Wikipedia-API==0.6.0
+wldhx.yadisk-direct==0.0.6
+xxhash==3.4.1
+yarl==1.9.4
+zipp==3.18.1
+zstandard==0.23.0