add requirements

requirements.txt

@@ -1,13 +1,380 @@
+absl-py==1.4.0
 accelerate==0.21.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
+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
 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
-transformers==4.30.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
 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