diff --git a/.github/actions/nlp_template_test/action.yml b/.github/actions/nlp_template_test/action.yml index ce39d3a..cb904ff 100644 --- 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 index 8630f12..acb3bff 100644 --- 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 index f7f9817..2b53a27 100644 --- a/template/requirements.txt +++ b/template/requirements.txt @@ -1,4 +1,3 @@ torchvision -accelerate gradio zenml[server]>=0.56.3