split out training packages

2026-03-21 05:45:45 +00:00 · 2025-05-17 20:55:43 +03:00
parent 7a5718bbfb
commit 508fc67bad
1 changed files with 7 additions and 0 deletions
--- a/setup.py
+++ b/setup.py
@@ -24,4 +24,11 @@ setup(
    package_data={
        "acestep.models.lyrics_utils": ["vocab.json"],  # Specify the relative path to vocab.json
    },
+    extras_require={
+        "train": [
+            "peft",
+            "tensorboard",
+            "tensorboardX"
+        ]
+    },
 )