remove training requirements to reduce size

metastable-studio · Apr 26, 2024 · 80f6f9a · 80f6f9a
1 parent f5d68d3
commit 80f6f9a
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 13 deletions.
diff --git 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
+torch
diff --git 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
+torch
diff --git a/requirements.directml.txt b/requirements.directml.txt
@@ -1,3 +1,2 @@
 -r requirements.txt
-torch-directml
-onnxruntime-directml
+torch-directml
diff --git 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
diff --git 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
+transformers