diff --git a/.github/workflows/train_workflow.yml b/.github/workflows/train_workflow.yml
index 1b7d44b..171fcd0 100644
--- a/.github/workflows/train_workflow.yml
+++ b/.github/workflows/train_workflow.yml
@@ -13,7 +13,7 @@ jobs:
     steps:
       - name: Install dependencies
         run: |
-          pip install -r ci-requirements.txt
+          pip install numpy torch
 
       - name: Create and run training script
         run: |