Update setup.py

831dc91a · Nuomanzz · 5d607345 · 831dc91a
Commit 831dc91a authored Jan 25, 2025 by Nuomanzz
Hide whitespace changes
Inline Side-by-side

Showing with 7 additions and 7 deletions

setup.py setup.py +7 -7

No files found.
--- a/setup.py
+++ b/setup.py
@@ -6,14 +6,14 @@ setup(
    version="0.1.0",
    packages=["tangoflux"],
    install_requires=[
-        "torch==2.4.0",
-        "torchaudio==2.4.0",
-        "torchlibrosa==0.1.0",
-        "torchvision==0.19.0",
-        "transformers==4.44.0",
+        "torch",
+        "torchaudio",
+        "torchlibrosa",
+        "torchvision==",
+        "transformers==",
        "diffusers==0.30.0",
-        "accelerate==0.34.2",
-        "datasets==2.21.0",
+        "accelerate",
+        "datasets",
        "librosa",
        "tqdm",
        "wandb",