updated reqs

requirements.txt

@@ -1,22 +1,166 @@
-git+https://github.com/huggingface/diffusers.git #diffusers==0.16.0.dev0
--f https://storage.googleapis.com/jax-releases/libtpu_releases.html
-jax[tpu]==0.4.5
+absl-py==1.4.0
+accelerate==0.16.0
+addict==2.4.0
+aiofiles==23.1.0
+aiohttp==3.8.4
+aiosignal==1.3.1
+albumentations==1.3.0
+altair==4.2.2
+antlr4-python3-runtime==4.9.3
+anyio==3.6.2
+async-timeout==4.0.2
+attrs==23.1.0
+basicsr==1.4.2
+beautifulsoup4==4.12.2
+braceexpand==0.1.7
+bs4==0.0.1
+cached-property==1.5.2
+cachetools==5.3.0
+certifi==2022.12.7
+charset-normalizer==3.1.0
+chex==0.1.7
+click==8.1.3
+coloredlogs==15.0.1
+contourpy==1.0.7
+cycler==0.11.0
+datasets==2.11.0
+decorator==4.4.2
+decord==0.6.0
+diffusers==0.16.0.dev0
+dill==0.3.6
+dm-tree==0.1.8
+einops==0.6.0
+entrypoints==0.4
+etils==1.2.0
+fastapi==0.95.1
+ffmpy==0.3.0
+filelock==3.11.0
+flatbuffers==23.3.3
 flax==0.6.7
-transformers
-chex
-datasets
-optax
-orbax
-ftfy
-tensorboard
-Jinja2
---extra-index-url https://download.pytorch.org/whl/cpu
-torch
-torchvision
-gradio
-einops
-imageio
-decord
-opencv-python
-scikit-image
-bs4
+fonttools==4.39.3
+frozenlist==1.3.3
+fsspec==2023.4.0
+ftfy==6.1.1
+future==0.18.3
+google-auth==2.17.3
+google-auth-oauthlib==1.0.0
+gradio==3.23.0
+grpcio==1.54.0
+h11==0.14.0
+httpcore==0.17.0
+httpx==0.24.0
+huggingface-hub==0.13.4
+humanfriendly==10.0
+idna==3.4
+imageio==2.9.0
+imageio-ffmpeg==0.4.2
+importlib-metadata==6.5.0
+importlib-resources==5.12.0
+invisible-watermark==0.1.5
+jax==0.4.5
+jaxlib==0.4.4
+Jinja2==3.1.2
+joblib==1.2.0
+jsonschema==4.17.3
+kiwisolver==1.4.4
+kornia==0.6.0
+linkify-it-py==2.0.0
+lmdb==1.4.1
+Markdown==3.4.3
+markdown-it-py==2.2.0
+MarkupSafe==2.1.2
+matplotlib==3.7.1
+mdit-py-plugins==0.3.3
+mdurl==0.1.2
+moviepy==1.0.3
+mpmath==1.3.0
+msgpack==1.0.5
+multidict==6.0.4
+multiprocess==0.70.14
+nest-asyncio==1.5.6
+networkx==3.1
+numpy==1.24.1
+oauthlib==3.2.2
+omegaconf==2.3.0
+onnx==1.13.1
+onnxruntime==1.14.1
+open-clip-torch==2.16.0
+opencv-contrib-python==4.7.0.72
+opencv-python==4.7.0.72
+opencv-python-headless==4.7.0.72
+opt-einsum==3.3.0
+optax==0.1.4
+orbax==0.1.7
+orjson==3.8.10
+packaging==23.1
+pandas==2.0.0
+Pillow==9.4.0
+pkgutil-resolve-name==1.3.10
+prettytable==3.6.0
+proglog==0.1.10
+protobuf==3.20.3
+psutil==5.9.5
+pyarrow==11.0.0
+pyasn1==0.4.8
+pyasn1-modules==0.2.8
+pydantic==1.10.7
+pyDeprecate==0.3.1
+pydub==0.25.1
+Pygments==2.15.0
+pyparsing==3.0.9
+pyrsistent==0.19.3
+python-dateutil==2.8.2
+python-multipart==0.0.6
+pytorch-lightning==1.5.0
+pytz==2023.3
+PyWavelets==1.4.1
+PyYAML==6.0
+qudida==0.0.4
+regex==2023.3.23
+requests==2.28.2
+requests-oauthlib==1.3.1
+responses==0.18.0
+rich==13.3.4
+rsa==4.9
+safetensors==0.2.7
+scikit-image==0.19.3
+scikit-learn==1.2.2
+scipy==1.10.1
+semantic-version==2.10.0
+sentencepiece==0.1.98
+six==1.16.0
+sniffio==1.3.0
+soupsieve==2.4.1
+starlette==0.26.1
+sympy==1.11.1
+tb-nightly==2.13.0a20230419
+tensorboard==2.12.2
+tensorboard-data-server==0.7.0
+tensorboard-plugin-wit==1.8.1
+tensorboardX==2.6
+tensorstore==0.1.35
+test-tube==0.7.5
+threadpoolctl==3.1.0
+tifffile==2023.4.12
+timm==0.6.12
+tokenizers==0.13.3
+tomesd==0.1.2
+toolz==0.12.0
+# torch==1.13.1+cpu
+torchmetrics==0.6.0
+# torchvision==0.14.1+cpu
+tqdm==4.64.1
+transformers==4.28.1
+typing-extensions==4.5.0
+tzdata==2023.3
+uc-micro-py==1.0.1
+urllib3==1.26.15
+uvicorn==0.21.1
+wcwidth==0.2.6
+webdataset==0.2.5
+websockets==11.0.2
+Werkzeug==2.2.3
+xxhash==3.2.0
+yapf==0.32.0
+yarl==1.8.2
+zipp==3.15.0