test

requirements.txt

@@ -1,12 +1,228 @@
+absl-py==1.4.0
 accelerate @ git+https://github.com/huggingface/accelerate.git@97d2168e5953fe7373a06c69c02c5a00a84d5344
--i https://pypi.org/simple/
-bitsandbytes
+-e git+https://github.com/NVIDIA/ai-assisted-annotation-client.git@d9426dfaa8023d0ebcaf468ad0d22428e9e7dc18#egg=aiaa_client
+aiofiles==23.2.1
+aiohttp==3.8.4
+aiosignal==1.3.1
+albumentations==1.3.1
+altair==5.2.0
+aniso8601==9.0.1
+annotated-types==0.5.0
+anyio==3.7.1
+appdirs==1.4.4
+asttokens==2.2.1
+async-timeout==4.0.2
+attrs==23.1.0
+backcall==0.2.0
+bitsandbytes==0.39.0
+bleach==6.0.0
+blinker==1.6.2
+Brotli @ file:///tmp/abs_ecyw11_7ze/croots/recipe/brotli-split_1659616059936/work
+cachetools==5.3.1
+certifi==2023.5.7
+charset-normalizer==3.1.0
+click==8.1.3
+cmake==3.26.4
+colorama==0.4.6
+comm==0.1.3
+contourpy==1.1.1
+cycler==0.11.0
+dataclasses-json==0.6.0
 datasets==2.17.1
+debugpy==1.6.7
+decorator==5.1.1
+dill==0.3.6
+docker-pycreds==0.4.0
+dominate==2.8.0
+efficientnet-pytorch==0.7.1
+et-xmlfile==1.1.0
+evaluate==0.4.0
+exceptiongroup==1.1.3
+executing==1.2.0
+fastapi==0.110.0
+ffmpy==0.3.2
+filelock==3.12.2
+Flask==2.3.2
+Flask-Cors==3.0.10
+flask-restx==1.1.0
+fonttools==4.42.1
+frozenlist==1.3.3
+fsspec==2023.6.0
+gguf==0.6.0
+gitdb==4.0.10
+GitPython==3.1.32
+gmpy2 @ file:///tmp/build/80754af9/gmpy2_1645455533097/work
+google-auth==2.22.0
+google-auth-oauthlib==1.0.0
+gradio==4.23.0
+gradio_client==0.14.0
+greenlet==2.0.2
+grpcio==1.57.0
+h11==0.14.0
+h5py==3.10.0
+httpcore==1.0.4
+httpx==0.27.0
 huggingface-hub==0.20.3
+idna==3.4
+imageio==2.31.3
+importlib_resources==6.4.0
+ipykernel==6.23.2
+ipython==8.14.0
+itsdangerous==2.1.2
+jedi==0.18.2
+Jinja2==3.1.2
+joblib==1.2.0
+jsonpatch==1.33
+jsonpointer==2.4
+jsonschema==4.17.3
+jupyter_client==8.2.0
+jupyter_core==5.3.0
+kaggle==1.5.16
+kiwisolver==1.4.5
+langchain==0.0.300
+langsmith==0.0.40
+lazy_loader==0.3
+lit==16.0.6
+Markdown==3.4.4
+markdown-it-py==3.0.0
+MarkupSafe==2.1.3
+marshmallow==3.20.1
+matplotlib==3.8.3
+matplotlib-inline==0.1.6
+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==1.3.0
+multidict==6.0.4
+multiprocess==0.70.14
+munch==4.0.0
+mypy-extensions==1.0.0
+nest-asyncio==1.5.6
+networkx==3.1
+nibabel==5.1.0
+numexpr==2.8.6
+numpy==1.25.0
+nvidia-cublas-cu11==11.10.3.66
+nvidia-cublas-cu12==12.1.3.1
+nvidia-cuda-cupti-cu11==11.7.101
+nvidia-cuda-cupti-cu12==12.1.105
+nvidia-cuda-nvrtc-cu11==11.7.99
+nvidia-cuda-nvrtc-cu12==12.1.105
+nvidia-cuda-runtime-cu11==11.7.99
+nvidia-cuda-runtime-cu12==12.1.105
+nvidia-cudnn-cu11==8.5.0.96
+nvidia-cudnn-cu12==8.9.2.26
+nvidia-cufft-cu11==10.9.0.58
+nvidia-cufft-cu12==11.0.2.54
+nvidia-curand-cu11==10.2.10.91
+nvidia-curand-cu12==10.3.2.106
+nvidia-cusolver-cu11==11.4.0.1
+nvidia-cusolver-cu12==11.4.5.107
+nvidia-cusparse-cu11==11.7.4.91
+nvidia-cusparse-cu12==12.1.0.106
+nvidia-nccl-cu11==2.14.3
+nvidia-nccl-cu12==2.19.3
+nvidia-nvjitlink-cu12==12.3.101
+nvidia-nvtx-cu11==11.7.91
+nvidia-nvtx-cu12==12.1.105
+openai==0.28.0
+opencv-python==4.8.1.78
+opencv-python-headless==4.8.1.78
+openpyxl==3.1.2
+orjson==3.9.15
+packaging==23.1
+pandas==2.0.2
+parso==0.8.3
+pathtools==0.1.2
 peft==0.8.2
-tokenizers==0.15.2 
-torch==2.2.1+cu118 
-torchaudio==2.2.1+cu118 
-torchvision==0.17.1+cu118 
-transformers==4.38.1
-trl==0.7.11
+pexpect==4.8.0
+pickleshare==0.7.5
+Pillow==9.5.0
+platformdirs==3.5.3
+pretrainedmodels==0.7.4
+prompt-toolkit==3.0.38
+protobuf==4.24.1
+psutil==5.9.5
+ptyprocess==0.7.0
+pure-eval==0.2.2
+pyarrow==12.0.1
+pyarrow-hotfix==0.6
+pyasn1==0.5.0
+pyasn1-modules==0.3.0
+pydantic==2.6.4
+pydantic_core==2.16.3
+pydub==0.25.1
+Pygments==2.15.1
+pyparsing==3.1.2
+pyrsistent==0.19.3
+PySocks @ file:///home/builder/ci_310/pysocks_1640793678128/work
+python-dateutil==2.8.2
+python-dotenv==1.0.0
+python-multipart==0.0.9
+python-slugify==8.0.1
+pytz==2023.3
+PyWavelets==1.4.1
+PyYAML==6.0
+pyzmq==25.1.0
+qudida==0.0.4
+regex==2023.6.3
+requests==2.31.0
+requests-oauthlib==1.3.1
+responses==0.18.0
+rich==13.7.1
+rsa==4.9
+ruff==0.3.4
+safetensors==0.4.2
+scikit-image==0.21.0
+scikit-learn==1.2.2
+scipy==1.10.1
+segmentation-models-pytorch==0.3.3
+semantic-version==2.10.0
+sentencepiece==0.1.99
+sentry-sdk==1.29.2
+setproctitle==1.3.2
+shellingham==1.5.4
+SimpleITK==2.3.0
+six==1.16.0
+smmap==5.0.0
+sniffio==1.3.0
+soynlp==0.0.493
+SQLAlchemy==2.0.21
+stack-data==0.6.2
+starlette==0.36.3
+sympy==1.12
+tenacity==8.2.3
+tensorboard==2.14.0
+tensorboard-data-server==0.7.1
+text-unidecode==1.3
+threadpoolctl==3.1.0
+tifffile==2023.9.18
+timm==0.9.2
+tokenizers==0.15.2
+tomlkit==0.12.0
+toolz==0.12.1
+torch==2.2.1
+torchaudio==2.2.1
+torchvision==0.17.1
+tornado==6.3.2
+tqdm==4.65.0
+traitlets==5.9.0
+transformers==4.39.1
+triton==2.2.0
+typer==0.10.0
+typing-inspect==0.9.0
+typing_extensions==4.10.0
+tzdata==2023.3
+urllib3==2.0.3
+uvicorn==0.29.0
+visdom==0.2.4
+wandb==0.15.9
+wcwidth==0.2.6
+webencodings==0.5.1
+websocket-client==1.6.3
+websockets==11.0.3
+Werkzeug==2.3.6
+xformers==0.0.22.post7
+xxhash==3.2.0
+yarl==1.9.2