accelerate unused

zenml-io · Sep 23, 2024 · 14d0f52 · 14d0f52
1 parent cf021ce
commit 14d0f52
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 3 deletions.
diff --git a/.github/actions/nlp_template_test/action.yml b/.github/actions/nlp_template_test/action.yml
@@ -82,7 +82,7 @@ runs:
         UV_HTTP_TIMEOUT: '1500'
       run: |
         uv pip install --system -r ./local_checkout/all-requirements.txt
-        uv pip install --system accelerate torchvision
+        uv pip install --system torchvision
 
     - name: Run pytests
       shell: bash

diff --git a/template/config.yaml b/template/config.yaml
@@ -26,7 +26,6 @@ settings:
       - mlflow
       - discord
     requirements:
-      - accelerate
       - zenml[server]
 
 extra:

diff --git a/template/requirements.txt b/template/requirements.txt
@@ -1,4 +1,3 @@
 torchvision
-accelerate
 gradio
 zenml[server]>=0.56.3
-Original file line number
+Diff line change
@@ Expand Up / @@ -26,7 +26,6 @@ settings: @@
           - mlflow
           - discord
         requirements:
-          - accelerate
           - zenml[server]
     extra:
@@ Expand Down @@