Create requirements.txt

requirements.txt

@@ -0,0 +1,25 @@
+# --------- pytorch --------- #
+
+lightning==2.5.5
+torch==2.8.0
+torchmetrics==1.8.2
+torchvision==0.23.0
+transformers==4.56.2
+datasets==4.1.1
+tokenizers==0.22.1
+modelscope
+# --------- hydra --------- #
+hydra-core==1.3.2
+hydra-colorlog==1.2.0
+hydra-optuna-sweeper==1.2.0
+
+# --------- loggers --------- #
+tensorboard==2.20.0
+
+# --------- others --------- #
+rootutils       # standardizing the project root setup
+pre-commit      # hooks for applying linters on commit
+rich            # beautiful text formatting in terminal
+pytest          # tests
+# sh            # for running bash commands in some tests (linux/macos only)
+gpustat