diff --git a/.github/workflows/train_workflow.yml b/.github/workflows/train_workflow.yml index 61d61fc..8a26621 100644 --- a/.github/workflows/train_workflow.yml +++ b/.github/workflows/train_workflow.yml @@ -17,7 +17,6 @@ jobs: - name: Install dependencies run: | - python -m pip install --upgrade pip pip install numpy pip install torch --index-url https://download.pytorch.org/whl/rocm6.2