diff --git a/requirements.txt b/requirements.txt new file mode 100755 index 0000000..6ab00ee --- /dev/null +++ b/requirements.txt @@ -0,0 +1,9 @@ +pillow +numpy==1.26.0 +transformers==4.28.1 +pytorch-lightning==1.8.5 +datasets +sentencepiece +--find-links https://download.pytorch.org/whl/torch_stable.html +torch==1.9.0+cu111 +nltk==3.7