envs

requirements.txt

@@ -1,235 +1,25 @@
-absl-py==2.1.0
-accelerate==0.23.0
-addict==2.4.0
-aenum==3.1.15
-aiofiles==23.2.1
-aiohttp==3.8.5
-aiosignal==1.3.1
-albumentations==1.3.0
-altair==4.2.2
-annotated-types==0.7.0
-antlr4-python3-runtime==4.9.3
-anyio==3.7.1
-argcomplete==3.4.0
-asttokens==2.4.1
-astunparse==1.6.3
-async-timeout==4.0.3
-attrs==23.2.0
-audioread==3.0.1
-av==11.0.0
-av2==0.2.1
-backcall==0.2.0
-basicsr==1.4.2
-beautifulsoup4==4.12.3
-bitsandbytes==0.41.1
-bleach==6.1.0
-blendmodes==2022
-blessed==1.20.0
-blinker==1.8.2
-braceexpand==0.1.7
-cachetools==4.2.4
-certifi==2024.6.2
-cffi==1.16.0
-chardet==5.2.0
-charset-normalizer==3.3.2
-clean-fid==0.1.35
-click==8.1.7
-clip @ git+https://github.com/openai/CLIP.git@d50d76daa670286dd6cacf3bcd80b5e4823fc8e1
-colored==2.2.4
-coloredlogs==15.0.1
-colorlog==6.8.2
-comm==0.2.2
-contourpy==1.2.1
-cos-python-sdk-v5==1.9.30
-coscmd==1.8.6.30
-crcmod==1.7
-cryptography==41.0.3
-cssselect2==0.7.0
-cuda-python==12.4.0
-cycler==0.12.1
-Cython==3.0.10
-dadaptation==3.1
-DateTime==5.5
-debugpy==1.8.1
-decorator==4.4.2
-deepdanbooru==1.0.2
-defusedxml==0.7.1
-deprecation==2.1.0
-diffusers==0.28.0
-dill==0.3.8
-discord-webhook==1.3.1
-diskcache==5.6.3
-distlib==0.3.8
-dnspython==2.6.1
-docopt==0.6.2
-dulwich==0.22.1
-easydict==1.13
-easygui==0.98.3
-einops==0.4.1
-email_validator==2.2.0
-entrypoints==0.4
-exceptiongroup==1.2.1
-executing==2.0.1
-face-alignment==1.3.5
-facexlib==0.3.0
-fairscale==0.4.13
-fastapi==0.94.1
-fastapi-cli==0.0.4
-fastjsonschema==2.20.0
-ffmpy==0.3.2
-filelock==3.13.1
-filterpy==1.4.5
-flatbuffers==24.3.25
-font-roboto==0.0.1
-fonts==0.0.3
-fonttools==4.53.0
-freetype-py==2.4.0
-frozenlist==1.4.1
-fsspec==2024.5.0
-ftfy==6.1.1
-future==1.0.0
-fuzzywuzzy==0.18.0
-fvcore==0.1.5.post20221221
-gast==0.4.0
-gdown==4.5.3
-gfpgan==1.3.8
-gitdb==4.0.11
-GitPython==3.1.32
-google-auth==1.35.0
-google-auth-oauthlib==0.4.6
-google-pasta==0.2.0
-gpustat==1.1.1
-GPUtil==1.4.0
-gradio==3.50.2
-gradio_client==0.6.1
-grpcio==1.64.0
-h11==0.12.0
-h5py==3.11.0
-hsluv==5.0.4
-httpcore==0.15.0
-httptools==0.6.1
-httpx==0.24.1
-huggingface-hub==0.23.4
-humanfriendly==10.0
-idna==3.7
-imageio==2.31.6
-imageio-ffmpeg==0.4.8
-importlib_metadata==7.1.0
-importlib_resources==6.4.0
-inflection==0.5.1
-insightface==0.7.3
-install==1.3.5
-invisible-watermark==0.2.0
-iopath==0.1.10
-ipdb==0.13.13
-IProgress==0.4
-ipykernel==6.29.4
-ipython==8.12.3
-ipywidgets==8.0.3
-jax==0.4.28
-jedi==0.19.1
-Jinja2==3.1.3
-joblib==1.1.0
-jsonmerge==1.8.0
-jsonschema==4.22.0
-jsonschema-specifications==2023.12.1
-jupyter_client==8.6.2
-jupyter_core==5.7.2
-jupyterlab_pygments==0.3.0
-jupyterlab_widgets==3.0.11
-keras==2.11.0
-kiwisolver==1.4.5
-kornia==0.6.8
-lark==1.1.2
-lazy_loader==0.4
-libclang==18.1.1
-librosa==0.9.2
-lightning-utilities==0.11.3.post0
-lion-pytorch==0.1.4
-llvmlite==0.41.1
-lmdb==1.5.1
-locon==0.0.4
-loguru==0.6.0
-lxml==4.9.3
-Markdown==3.6
-markdown-it-py==3.0.0
-MarkupSafe==2.1.5
-matplotlib==3.6.2
-matplotlib-inline==0.1.7
-mdurl==0.1.2
-mediapipe==0.10.3
-mistune==3.0.2
-ml-dtypes==0.4.0
-modelcards==0.1.6
-moviepy==1.0.2
-mpmath==1.3.0
-multidict==6.0.5
-multiprocess==0.70.16
-nbclient==0.10.0
-nbconvert==7.16.4
-nbformat==5.10.4
-nest-asyncio==1.6.0
-networkx==3.1
-ninja==1.11.1.1
-nox==2024.4.15
-numba==0.58.1
-numpy==1.26.2
-nvidia-cublas-cu12==12.1.3.1
-nvidia-cuda-cupti-cu12==12.1.105
-nvidia-cuda-nvrtc-cu12==12.1.105
-nvidia-cuda-runtime-cu12==12.1.105
-nvidia-cudnn-cu12==8.9.2.26
-nvidia-cufft-cu12==11.0.2.54
-nvidia-curand-cu12==10.3.2.106
-nvidia-cusolver-cu12==11.4.5.107
-nvidia-cusparse-cu12==12.1.0.106
-nvidia-ml-py==12.555.43
-nvidia-nccl-cu12==2.20.5
-nvidia-nvjitlink-cu12==12.1.105
-nvidia-nvtx-cu12==12.1.105
-nvidia-pyindex==1.0.9
-nvtx==0.2.10
-oauthlib==3.2.2
-omegaconf==2.2.3
-open_clip_torch @ git+https://github.com/mlfoundations/open_clip.git@bb6e834e9c70d9c27d0dc3ecedeebeaeb1ffad6b
-opencv-contrib-python==4.10.0.84
-opencv-python-headless==4.10.0.84
-packaging==24.0
-pandas==2.2.2
-peft==0.11.1
-python-dateutil==2.9.0.post0
-python-dotenv==1.0.1
-python-multipart==0.0.9
-pytorch-lightning==1.9.4
-pytorch_optimizer==2.12.0
-PyWavelets==1.6.0
-PyYAML==6.0.1
-regex==2024.5.15
-rembg==2.0.50
-reportlab==4.2.2
-requests==2.28.2
-safetensors==0.4.2
-scikit-image==0.21.0
-scikit-learn==1.1.3
-scipy==1.10.1
-six==1.16.0
-toolz==0.12.1
-torch==2.3.0
-torchaudio==2.3.0
-torchvision==0.18.0
-tqdm==4.65.0
-transformers==4.32.1
-transforms3d==0.4.1
-trimesh==4.0.4
-triton==2.3.0
-typer==0.12.3
-typing_extensions==4.12.1
-widgetsnbextension==4.0.11
-wrapt==1.14.1
-xformers==0.0.26.post1
-xmltodict==0.13.0
-yacs==0.1.8
-yapf==0.40.2
-yarg==0.1.9
-yarl==1.9.4
-zipp==3.19.1
+torch
+torchvision
+torchaudio
+transformers>=4.25.1
+gradio==3.50.0
+ftfy
+tensorboard
+datasets
+Pillow==9.5.0 
+opencv-python 
+imgaug 
+accelerate==0.20.3
+image-reward
+hpsv2
+torchmetrics
+open-clip-torch
+clip
+av2
+peft
+imageio-ffmpeg
+scipy
+tqdm
+einops
+diffusers==0.29.2
+omegaconf