split out training packages

2025-05-17 20:55:43 +03:00
parent 7a5718bbfb
commit 508fc67bad
1 changed files with 7 additions and 0 deletions
@@ -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"
        ]
    },
 )