From 80f6f9af7c044ccea91df5b7dde2e139a4e71812 Mon Sep 17 00:00:00 2001 From: Mat Sz Date: Fri, 26 Apr 2024 18:59:59 +0200 Subject: [PATCH] remove training requirements to reduce size --- requirements.cpu.txt | 3 +-- requirements.cuda.txt | 3 +-- requirements.directml.txt | 3 +-- requirements.rocm.txt | 3 +-- requirements.txt | 6 +----- 5 files changed, 5 insertions(+), 13 deletions(-) diff --git a/requirements.cpu.txt b/requirements.cpu.txt index d775cc7..58ad573 100644 --- a/requirements.cpu.txt +++ b/requirements.cpu.txt @@ -1,4 +1,3 @@ --extra-index-url https://download.pytorch.org/whl/nightly/cpu -r requirements.txt -torch -onnxruntime \ No newline at end of file +torch \ No newline at end of file diff --git a/requirements.cuda.txt b/requirements.cuda.txt index 1dcc51c..d876fe9 100644 --- a/requirements.cuda.txt +++ b/requirements.cuda.txt @@ -1,5 +1,4 @@ --extra-index-url https://download.pytorch.org/whl/cu121 --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/ -r requirements.txt -torch -onnxruntime-gpu \ No newline at end of file +torch \ No newline at end of file diff --git a/requirements.directml.txt b/requirements.directml.txt index 97cc5ca..7c1442b 100644 --- a/requirements.directml.txt +++ b/requirements.directml.txt @@ -1,3 +1,2 @@ -r requirements.txt -torch-directml -onnxruntime-directml \ No newline at end of file +torch-directml \ No newline at end of file diff --git a/requirements.rocm.txt b/requirements.rocm.txt index de7c9a9..ce329dc 100644 --- a/requirements.rocm.txt +++ b/requirements.rocm.txt @@ -1,4 +1,3 @@ --extra-index-url https://download.pytorch.org/whl/nightly/rocm6.0 -r requirements.txt -torch -onnxruntime +torch \ No newline at end of file diff --git a/requirements.txt b/requirements.txt index 8c4e018..74e0b27 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,13 +2,9 @@ torchvision torchsde einops safetensors -accelerate pyyaml Pillow scipy tqdm psutil -transformers -diffusers[torch] -voluptuous -onnx \ No newline at end of file +transformers \ No newline at end of file