upd requiremnts

requirements.txt

@@ -1,27 +1,213 @@
---extra-index-url https://download.pytorch.org/whl/cu118
-pip
-einops
-fastapi
+aac-metrics==0.5.3
+absl-py==2.0.0
+accelerate==0.24.0
+aiofiles==23.2.1
+aiohttp==3.8.6
+aiosignal==1.3.1
+albumentations==1.3.1
+altair==5.1.2
+anyio==3.7.1
+appdirs==1.4.4
+async-timeout==4.0.3
+attrs==23.1.0
+audioread==3.0.1
+auto-gptq==0.5.1
+beautifulsoup4==4.12.2
+bitsandbytes==0.41.2.post2
+braceexpand==0.1.7
+brotlipy==0.7.0
+cachetools==5.3.1
+cchardet==2.1.7
+certifi @ file:///croot/certifi_1690232220950/work/certifi
+cffi==1.16.0
+chardet==5.2.0
+charset-normalizer @ file:///tmp/build/80754af9/charset-normalizer_1630003229654/work
+click==8.1.7
+cmake==3.28.1
+coloredlogs==15.0.1
+contourpy==1.1.1
+cryptography @ file:///croot/cryptography_1694444244250/work
+cycler==0.12.1
+dataclasses==0.6
+datasets==2.14.7
+decorator==5.1.1
+deepspeed==0.9.5
+dill==0.3.7
+docker-pycreds==0.4.0
+einops==0.6.1
+einops-exts==0.0.4
+exceptiongroup==1.1.3
+ExifRead-nocycle==3.0.1
+fastapi==0.104.0
+ffmpy==0.3.1
+filelock==3.12.4
+fire==0.4.0
+flash-attn==2.3.3
+fonttools==4.43.1
+frozenlist==1.4.0
+fsspec==2023.10.0
+ftfy==6.1.3
+gdown==4.6.3
+gekko==1.0.6
+gitdb==4.0.11
+GitPython==3.1.40
+gmpy2 @ file:///tmp/build/80754af9/gmpy2_1645455533097/work
+google-auth==2.23.0
+google-auth-oauthlib==1.0.0
 gradio==3.35.2
-markdown2[all]
-numpy
-requests
-sentencepiece
-tokenizers>=0.12.1
-torch==2.0.1
-torchvision==0.15.2
-uvicorn
-wandb
-shortuuid
+gradio_client==0.2.9
+grpcio==1.58.0
+h11==0.14.0
+hjson==3.1.0
+httpcore==0.17.3
 httpx==0.24.0
-deepspeed==0.9.5
-peft==0.4.0
-transformers==4.36.0
-accelerate==0.21.0
-bitsandbytes==0.41.0
+huggingface-hub==0.20.3
+humanfriendly==10.0
+idna @ file:///croot/idna_1666125576474/work
+imageio==2.32.0
+img2dataset==1.42.0
+Jinja2 @ file:///croot/jinja2_1666908132255/work
+joblib==1.3.2
+jsonschema==4.19.1
+jsonschema-specifications==2023.7.1
+kiwisolver==1.4.5
+lazy_loader==0.3
+librosa==0.10.1
+lightning-utilities==0.10.1
+linkify-it-py==2.0.2
+lit==17.0.6
+llvmlite==0.41.0
+Markdown==3.4.4
+markdown-it-py==2.2.0
+markdown2==2.4.10
+MarkupSafe @ file:///opt/conda/conda-bld/markupsafe_1654597864307/work
+matplotlib==3.8.0
+mdit-py-plugins==0.3.3
+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
+msgpack==1.0.7
+multidict==6.0.4
+multiprocess==0.70.15
+networkx @ file:///croot/networkx_1690561992265/work
+ninja==1.11.1.1
+nltk==3.8.1
+numba==0.58.0
+numpy @ file:///croot/numpy_and_numpy_base_1695830428084/work/dist/numpy-1.26.0-cp310-cp310-linux_x86_64.whl#sha256=fc2732718bc9e06a7b702492cb4f5afffe9671083930452d894377bf563464a3
+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.18.1
+nvidia-nvjitlink-cu12==12.3.52
+nvidia-nvtx-cu11==11.7.91
+nvidia-nvtx-cu12==12.1.105
+oauthlib==3.2.2
+open-clip-torch==2.23.0
+opencv-python-headless==4.8.1.78
+optimum==1.14.1
+orjson==3.9.9
+packaging==23.2
+pandas==2.1.0
+pathtools==0.1.2
+peft==0.6.1
+Pillow @ file:///croot/pillow_1696580024257/work
+platformdirs==3.11.0
+pooch==1.7.0
+promise==2.3
+protobuf==3.20.3
+psutil==5.9.6
+py-cpuinfo==9.0.0
+pyarrow==10.0.1
+pyarrow-hotfix==0.6
+pyasn1==0.5.0
+pyasn1-modules==0.3.0
+pycocoevalcap @ git+https://github.com/salaniz/pycocoevalcap.git@a24f74c408c918f1f4ec34e9514bc8a76ce41ffd
+pycocotools==2.0.7
+pycparser==2.21
+pydantic==1.10.13
+pydub==0.25.1
+Pygments==2.16.1
+pyOpenSSL @ file:///croot/pyopenssl_1690223430423/work
+pyparsing==3.1.1
+PySocks @ file:///home/builder/ci_310/pysocks_1640793678128/work
+python-dateutil==2.8.2
+python-multipart==0.0.6
+pytz==2023.3.post1
+PyYAML @ file:///croot/pyyaml_1698096049011/work
+qudida==0.0.4
+referencing==0.30.2
+regex==2023.10.3
+requests @ file:///croot/requests_1690400202158/work
+requests-oauthlib==1.3.1
+robin @ file:///p/project/laionize/marianna/bakllava_original/Robin
+rouge==1.0.1
+rpds-py==0.10.6
+rsa==4.9
+safetensors==0.4.0
+scikit-image==0.22.0
 scikit-learn==1.2.2
+scipy==1.11.3
+semantic-version==2.10.0
+sentence-transformers==2.2.2
 sentencepiece==0.1.99
-einops==0.6.1
-einops-exts==0.0.4
+sentry-sdk==1.32.0
+setproctitle==1.3.3
+shortuuid==1.0.11
+six==1.16.0
+smmap==5.0.1
+sniffio==1.3.0
+soundfile==0.12.1
+soupsieve==2.5
+soxr==0.3.6
+starlette==0.27.0
+svgwrite==1.4.3
+sympy @ file:///croot/sympy_1668202399572/work
+tensorboard==2.14.0
+tensorboard-data-server==0.7.1
+termcolor==2.3.0
+threadpoolctl==3.2.0
+tifffile==2023.9.26
 timm==0.6.13
-gradio_client==0.2.9
+tokenizers==0.15.1
+toml==0.10.2
+toml-to-requirements==0.2.0
+toolz==0.12.0
+torch==2.0.1
+torchaudio==2.1.0
+torchlibrosa==0.0.2
+torchmetrics==1.3.0.post0
+torchvision==0.16.0
+tqdm==4.66.1
+transformers==4.37.0
+triton==2.0.0
+typing_extensions==4.8.0
+tzdata==2023.3
+uc-micro-py==1.0.2
+urllib3==1.26.16
+uvicorn==0.23.2
+wandb==0.12.21
+wavedrom==2.0.3.post3
+wcwidth==0.2.12
+webdataset==0.2.62
+websockets==12.0
+Werkzeug==2.3.7
+xxhash==3.4.1
+yarl==1.9.2