update requirements

requirements.txt

@@ -1,380 +1,14 @@
-absl-py==1.4.0
 accelerate==0.21.0
-addict==2.4.0
-aiobotocore==2.5.4
-aiofiles==23.2.1
-aiohttp==3.8.5
-aioitertools==0.11.0
-aiosignal==1.3.1
-albumentations==1.3.1
-aliyun-python-sdk-core==2.13.36
-aliyun-python-sdk-kms==2.16.1
-altair==5.1.2
-antlr4-python3-runtime==4.9.3
-anyio==3.7.1
-apex @ file:///home/chenchen/liming/code/apex
-appdirs==1.4.4
-argcomplete==3.1.1
-argon2-cffi==23.1.0
-argon2-cffi-bindings==21.2.0
-arrow==1.3.0
-asttokens==2.4.0
-async-lru==2.0.4
-async-timeout==4.0.3
-attrs==23.1.0
-Babel==2.13.1
-backcall==0.2.0
-beautifulsoup4==4.12.2
-bitsandbytes==0.41.1
-black==23.10.1
-bleach==6.1.0
-blessed==1.20.0
-blinker==1.6.3
-blis==0.7.11
-boto==2.49.0
-botocore==1.31.17
-braceexpand==0.1.7
-brotlipy==0.7.0
-cachetools==5.3.1
-catalogue==2.0.10
-cchardet==2.1.7
-certifi @ file:///croot/certifi_1690232220950/work/certifi
-cffi @ file:///tmp/abs_98z5h56wf8/croots/recipe/cffi_1659598650955/work
-cfgv==3.4.0
-charset-normalizer @ file:///tmp/build/80754af9/charset-normalizer_1630003229654/work
-chex==0.1.82
-chumpy==0.70
-clean-fid==0.1.35
-click==8.1.7
-clip @ git+https://github.com/openai/CLIP.git@a1d071733d7111c9c014f024669f959182114e33
-cloudpathlib==0.16.0
-cloudpickle==3.0.0
-cmake==3.27.7
-colorama==0.4.6
-comm==0.1.4
-confection==0.1.3
-contexttimer==0.3.3
-contourpy==1.1.0
-controlnet-aux==0.0.7
-crcmod==1.7
-cryptography @ file:///croot/cryptography_1689373676338/work
-cycler==0.11.0
-cymem==2.0.8
-Cython==3.0.0
-dataclasses==0.6
-datasets==2.17.1
-debugpy==1.8.0
-decorator==5.1.1
-decord==0.6.0
-deepspeed==0.10.2
-defusedxml==0.7.1
-detectron2 @ git+https://github.com/facebookresearch/detectron2.git@898507047cf441a1e4be7a729270961c401c4354
-diffusers==0.26.3
-dill==0.3.7
-distlib==0.3.7
-distro==1.8.0
-docker-pycreds==0.4.0
-docstring-parser==0.15
-einops==0.4.1
-etils==1.4.1
-exceptiongroup==1.1.3
-executing==2.0.0
-ExifRead-nocycle==3.0.1
-fairscale==0.4.13
-fastapi==0.104.0
-fasteners==0.18
-fastjsonschema==2.18.1
-ffmpy==0.3.1
-filelock==3.12.4
-fire==0.4.0
-flax==0.7.4
-fonttools==4.42.1
-fqdn==1.5.1
-frozenlist==1.4.0
-fsspec==2023.6.0
-ftfy==6.1.3
-fvcore==0.1.5.post20221221
-gcs-oauth2-boto-plugin==3.0
-gitdb==4.0.10
-GitPython==3.1.32
-gmpy2 @ file:///tmp/build/80754af9/gmpy2_1645455533097/work
-google-apitools==0.5.32
-google-auth==2.22.0
-google-auth-oauthlib==1.0.0
-google-reauth==0.1.1
-gpustat==1.1.1
-gradio==3.50.2
-gradio_client==0.6.1
-grpcio==1.57.0
-gsutil==5.25
-h11==0.14.0
-h5py==3.10.0
-hjson==3.1.0
-httpcore==0.18.0
-httplib2==0.20.4
-httpx==0.25.0
-huggingface-hub==0.21.2
-hydra-core==1.3.2
-identify==2.5.30
-idna @ file:///croot/idna_1666125576474/work
-image-reward==1.5
-imageio==2.31.2
-img2dataset==1.42.0
-importlib==1.0.4
-importlib-metadata==6.8.0
-importlib-resources==6.0.1
-iniconfig==2.0.0
-iopath==0.1.9
-ipykernel==6.26.0
-ipython==8.16.1
-isoduration==20.11.0
-jax==0.4.14
-jaxlib==0.4.14
-jedi==0.19.1
-Jinja2 @ file:///croot/jinja2_1666908132255/work
-jmespath==0.10.0
-joblib==1.3.2
-json-tricks==3.17.3
-json5==0.9.14
-jsonargparse==4.27.1
-jsonlines==4.0.0
-jsonpointer==2.4
-jsonschema==4.19.1
-jsonschema-specifications==2023.7.1
-jupyter-events==0.8.0
-jupyter-lsp==2.2.0
-jupyter_client==8.5.0
-jupyter_core==5.4.0
-jupyter_server==2.9.1
-jupyter_server_terminals==0.4.4
-jupyterlab==4.0.7
-jupyterlab-pygments==0.2.2
-jupyterlab_server==2.25.0
-kaggle==1.5.16
-kiwisolver==1.4.4
-kornia==0.7.0
-langcodes==3.3.0
-lazy_loader==0.3
-lightning==2.1.0
-lightning-utilities==0.9.0
-lit==17.0.3
-Markdown==3.4.4
-markdown-it-py==3.0.0
-MarkupSafe @ file:///opt/conda/conda-bld/markupsafe_1654597864307/work
-mat4py==0.6.0
-matplotlib==3.7.2
-matplotlib-inline==0.1.6
-mdurl==0.1.2
-megatron-lm @ git+https://github.com/huggingface/Megatron-LM.git@100b522bb8044d98413398f9e71563af15b83325
-mistune==3.0.2
-mkl-fft==1.3.1
-mkl-random @ file:///home/builder/ci_310/mkl_random_1641843545607/work
-mkl-service==2.4.0
-ml-dtypes==0.2.0
-mmcls==0.25.0
-mmcv==2.0.1
-mmdet==3.1.0
-mmengine==0.9.1
--e git+ssh://git@github.com/liming-ai/mmpose.git@eb4eb3829bb8c18c5ab92eae9691e520ed21887a#egg=mmpose
-mmpretrain==1.1.0
-mmsegmentation @ git+https://github.com/open-mmlab/mmsegmentation.git@cbf9af1e3f2eb50b58c8d783008f749ef62f2435
-model-index==0.1.11
-modelindex==0.0.2
-monotonic==1.6
-mpmath @ file:///croot/mpmath_1690848262763/work
-msgpack==1.0.5
-multidict==6.0.4
-multiprocess==0.70.15
-munkres==1.1.4
-murmurhash==1.0.10
-mypy-extensions==1.0.0
-nbclient==0.8.0
-nbconvert==7.9.2
-nbformat==5.9.2
-nest-asyncio==1.5.7
-networkx @ file:///croot/networkx_1690561992265/work
-ninja==1.11.1
-nltk==3.8.1
-nodeenv==1.8.0
-notebook_shim==0.2.3
-numpy==1.25.2
-nvidia-ml-py==12.535.108
-nvitop==1.3.1
-oauth2client==4.1.3
-oauthlib==3.2.2
-omegaconf==2.3.0
-open-clip-torch==2.22.0
-openai==0.28.0
-opencv-python==4.8.1.78
-opendatalab==0.0.10
-opendatasets==0.1.22
-openmim==0.3.9
-openxlab==0.0.28
-opt-einsum==3.3.0
-optax==0.1.7
-orbax-checkpoint==0.3.5
-ordered-set==4.1.0
-orjson==3.9.9
-oss2==2.17.0
-overrides==7.4.0
-packaging==23.1
-pandas==1.5.3
-pandocfilters==1.5.0
-panopticapi @ git+https://github.com/cocodataset/panopticapi.git@7bb4655548f98f3fedc07bf37e9040a992b054b0
-parso==0.8.3
-pathspec==0.11.2
-pathtools==0.1.2
-peft==0.5.0
-perceiver-io==0.11.0
-pexpect==4.8.0
-pickleshare==0.7.5
-pillow==10.2.0
-platformdirs==3.10.0
-plotly==5.17.0
-pluggy==1.3.0
-portalocker==2.8.2
-pre-commit==3.5.0
-preshed==3.0.9
-prettytable==3.8.0
-prometheus-client==0.17.1
-promise==2.3
-prompt-toolkit==3.0.39
-protobuf==3.20.3
-psutil==5.9.5
-ptyprocess==0.7.0
-pure-eval==0.2.2
-py-cpuinfo==9.0.0
-pyarrow==15.0.0
-pyarrow-hotfix==0.6
-pyasn1==0.5.0
-pyasn1-modules==0.3.0
-pybind11==2.11.1
-pycocoevalcap==1.2
-pycocotools==2.0.7
-pycparser @ file:///tmp/build/80754af9/pycparser_1636541352034/work
-pycryptodome==3.18.0
-pydantic==1.10.12
-pydeck==0.8.1b0
-pydub==0.25.1
-Pygments==2.16.1
-pyOpenSSL @ file:///croot/pyopenssl_1690223430423/work
-pyparsing==3.0.9
-pyre-extensions==0.0.29
-PySocks @ file:///home/builder/ci_310/pysocks_1640793678128/work
-pytest==7.4.3
-python-dateutil==2.8.2
-python-json-logger==2.0.7
-python-magic==0.4.27
-python-multipart==0.0.6
-python-slugify==8.0.1
-pytorch-lightning==2.0.9.post0
-pytorch-ranger==0.1.1
-pytz==2023.3
-pyu2f==0.1.5
-PyWavelets==1.4.1
-PyYAML==6.0.1
-pyzmq==25.1.1
-qudida==0.0.4
-referencing==0.30.2
-regex==2023.8.8
-requests==2.31.0
-requests-oauthlib==1.3.1
-retry-decorator==1.1.1
-rfc3339-validator==0.1.4
-rfc3986-validator==0.1.1
-rich==13.4.2
-rouge==1.0.1
-rpds-py==0.10.6
-rsa==4.7.2
-s3fs==2023.9.2
-safetensors==0.4.2
-# Editable install with no version control (salesforce-lavis==1.0.1)
--e /home/chenchen/liming/code/LAVIS
-scikit-image==0.21.0
-scikit-learn==1.3.0
-scipy==1.11.1
-seaborn==0.12.2
-segment-anything @ git+https://github.com/facebookresearch/segment-anything.git@6fdee8f2727f4506cfbbe553e23b895e27956588
-semantic-version==2.10.0
-Send2Trash==1.8.2
-sentence-transformers==2.2.2
-sentencepiece==0.1.99
-sentry-sdk==1.29.2
-setproctitle==1.3.2
-shapely==2.0.1
-shortuuid==1.0.11
-six @ file:///tmp/build/80754af9/six_1644875935023/work
-smart-open==6.4.0
-smmap==5.0.0
-sniffio==1.3.0
-soupsieve==2.5
-spacy==3.7.2
-spacy-legacy==3.0.12
-spacy-loggers==1.0.5
-srsly==2.4.8
-stack-data==0.6.3
-starlette==0.27.0
-streamlit==1.27.2
-sympy @ file:///croot/sympy_1668202399572/work
-tabulate==0.9.0
-tenacity==8.2.3
-tensorboard==2.16.2
-tensorboard-data-server==0.7.1
-tensorstore==0.1.43
-termcolor==2.3.0
-terminado==0.17.1
-terminaltables==3.1.10
-text-unidecode==1.3
-thinc==8.2.1
-threadpoolctl==3.2.0
-tifffile==2023.8.25
-timm==0.6.13
-tinycss2==1.2.1
-tokenizers==0.15.2
-toml==0.10.2
-tomli==2.0.1
-toolz==0.12.0
+controlnet_aux==0.0.6
+diffusers==0.18.2
+einops==0.6.1
+gradio==3.45.2
+huggingface-hub==0.16.4
+mediapipe==0.10.1
+opencv-python-headless==4.8.0.74
+safetensors==0.3.1
 torch==2.0.1
-torch-fidelity==0.3.0
-torch-optimizer==0.3.0
-torchaudio==2.0.2
-torcheval==0.0.7
-torcheval-nightly==2024.2.15
-torchmetrics==1.3.0
 torchvision==0.15.2
-tornado==6.3.3
-tqdm==4.65.2
-traitlets==5.11.2
-transformers==4.38.1
-triton==2.0.0
-typer==0.9.0
-types-python-dateutil==2.8.19.14
-typeshed-client==2.4.0
-typing-inspect==0.9.0
-typing_extensions==4.8.0
-tzdata==2023.3
-tzlocal==5.1
-uri-template==1.3.0
-urllib3 @ file:///croot/urllib3_1686163155763/work
-uvicorn==0.23.2
-validators==0.22.0
-virtualenv==20.24.5
-wandb==0.16.3
-wasabi==1.1.2
-watchdog==3.0.0
-wcwidth==0.2.13
-weasel==0.3.3
-webcolors==1.13
-webdataset==0.2.48
-webencodings==0.5.1
-websocket-client==1.6.4
-websockets==11.0.3
-Werkzeug==2.3.7
-wrapt==1.16.0
+transformers==4.30.2
 xformers==0.0.20
-xtcocotools==1.13
-xxhash==3.3.0
-yacs==0.1.8
-yapf==0.40.1
-yarl==1.9.2
-zipp==3.16.2
+kornia==0.7.0