diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..fda58f9 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,74 @@ +accelerate==0.30.1 +aiohttp==3.9.5 +aiosignal==1.3.1 +attrs==23.2.0 +certifi==2024.2.2 +charset-normalizer==3.3.2 +click==8.1.7 +contourpy==1.2.1 +cycler==0.12.1 +datasets==2.19.1 +diffusers==0.28.0 +dill==0.3.8 +docker-pycreds==0.4.0 +evaluate==0.4.2 +filelock==3.14.0 +fonttools==4.52.4 +frozenlist==1.4.1 +fsspec==2024.3.1 +gitdb==4.0.11 +gitpython==3.1.43 +huggingface-hub==0.23.2 +idna==3.7 +jinja2==3.1.4 +jupyter +kiwisolver==1.4.5 +markupsafe==2.1.5 +matplotlib==3.9.0 +mpmath==1.3.0 +multidict==6.0.5 +multiprocess==0.70.16 +networkx==3.3 +numpy==1.26.4 +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-nccl-cu12==2.20.5 +nvidia-nvjitlink-cu12==12.5.40 +nvidia-nvtx-cu12==12.1.105 +pandas==2.2.2 +git+https://github.com/huggingface/peft.git +pillow==10.3.0 +protobuf==4.25.3 +psutil==5.9.8 +pyarrow==16.1.0 +pyarrow-hotfix==0.6 +pyparsing==3.1.2 +python-dateutil==2.9.0.post0 +pytz==2024.1 +pyyaml==6.0.1 +regex==2024.5.15 +requests==2.32.3 +safetensors==0.4.3 +sentry-sdk==2.3.1 +setproctitle==1.3.3 +smmap==5.0.1 +sympy==1.12.1 +tokenizers==0.19.1 +torch==2.3.0 +torchvision==0.18.0 +tqdm==4.66.4 +transformers==4.41.1 +typing-extensions==4.12.0 +tzdata==2024.1 +urllib3==2.2.1 +wandb==0.17.0 +xxhash==3.4.1 +yarl==1.9.4 +zipp==3.19.0