fix ort ci

huggingface · Dec 10, 2024 · b8515f0 · b8515f0
1 parent 1bddc66
commit b8515f0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/test_cli_cuda_torch_ort.yaml b/.github/workflows/test_cli_cuda_torch_ort.yaml
@@ -45,7 +45,7 @@ jobs:
       - name: Install dependencies
         run: |
           pip install -e .[testing,torch-ort,peft]
-          pip install optimum@git+https://github.com/huggingface/optimum.git
+          pip install optimum@git+https://github.com/huggingface/optimum.git@fxi-ort-trainer
 
       - name: Run tests
         run: |
@@ -80,7 +80,7 @@ jobs:
       - name: Install dependencies
         run: |
           pip install -e .[testing,torch-ort,peft] 
-          pip install optimum@git+https://github.com/huggingface/optimum.git
+          pip install optimum@git+https://github.com/huggingface/optimum.git@fxi-ort-trainer
 
       - name: Run tests
         run: |