add requirements.txt

sbartlett97 · Dec 5, 2024 · c55af5d · c55af5d
1 parent e197aa1
commit c55af5d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/.github/workflows/pylint.yml b/.github/workflows/pylint.yml
@@ -17,6 +17,7 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
+        python -m pip install -r requirements.txt
         pip install pylint
     - name: Analysing the code with pylint
       run: |

diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,4 @@
+torch~=2.5.1
+transformers~=4.46.3
+datasets~=3.1.0
+matplotlib~=3.9.3