add requirements.txt

ckpt/best.pt.tar

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:298e4fccf3baacd8623574e7a767d22198f3ddb0c35cdb17e71e03dd4edf0fc5
-size 11826355083

requirements.txt

@@ -0,0 +1,18 @@
+accelerate==1.1.0
+aiohttp==3.10.5
+datasets==3.0.1
+deepspeed==0.15.1
+einops==0.8.0
+huggingface-hub==0.25.1
+numpy==1.23.5
+pandas==2.2.3
+pillow==10.4.0
+scikit-learn==1.5.2
+scipy==1.13.1
+torch==2.4.1
+torchaudio==2.4.1
+torchmetrics==1.4.2
+transformers==4.47.1
+tqdm==4.66.5
+torchao==0.5.0
+torchtune==0.3.1