SongGeneration-Studio/install.js at main · BazedFrog/SongGeneration-Studio · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
module.exports = {
  requires: {
    bundle: "ai"
  },
  run: [
    // 1. Clone the SongGeneration repo (contains codeclm/, web/, etc.)
    {
      method: "shell.run",
      params: {
        env: {
          GIT_LFS_SKIP_SMUDGE: "1"
         },
        message: [
          "git clone https://github.com/tencent-ailab/SongGeneration app"
        ]
      }
    },
    // 2. Download new_prompt.pt (Tencent reached its quota - git clone with lfs fails)
    {
      method: "fs.download",
      params: {
        dir: "app/tools",
        uri: "https://github.com/tencent-ailab/SongGeneration/raw/refs/heads/main/tools/new_prompt.pt?download="
      }
    },
    // 3. Download model weights from HuggingFace (ckpt/, third_party/)
    {
      method: "hf.download",
      params: {
        path: "app",
        _: ["lglg666/SongGeneration-Runtime"],
        "local-dir": "."
      }
    },
    // 4. Override Tencent's requirements with our tested working versions
    // (Tencent's GitHub has newer INCOMPATIBLE versions!)
    { method: "fs.copy", params: { src: "requirements.txt", dest: "app/requirements.txt" } },
    { method: "fs.copy", params: { src: "requirements_nodeps.txt", dest: "app/requirements_nodeps.txt" } },

    // 5. Install Python dependencies
    {
      method: "shell.run",
      params: {
        venv: "env",
        path: "app",
        message: [
          "uv pip install -r requirements.txt",
          "uv pip install -r requirements_nodeps.txt --no-deps"
        ]
      }
    },
    // 6. Install PyTorch (cross-platform via torch.js)
    {
      method: "script.start",
      params: {
        uri: "torch.js",
        params: {
          path: "app",
          venv: "env"
        }
      }
    },
    // 7. Sync custom Python files from root to app/
    { method: "fs.copy", params: { src: "main.py", dest: "app/main.py" } },
    { method: "fs.copy", params: { src: "generation.py", dest: "app/generation.py" } },
    { method: "fs.copy", params: { src: "model_server.py", dest: "app/model_server.py" } },
    { method: "fs.copy", params: { src: "models.py", dest: "app/models.py" } },
    { method: "fs.copy", params: { src: "gpu.py", dest: "app/gpu.py" } },
    { method: "fs.copy", params: { src: "config.py", dest: "app/config.py" } },
    { method: "fs.copy", params: { src: "schemas.py", dest: "app/schemas.py" } },
    { method: "fs.copy", params: { src: "sse.py", dest: "app/sse.py" } },
    { method: "fs.copy", params: { src: "timing.py", dest: "app/timing.py" } },
    // 8. Sync custom web files from root to app/
    { method: "fs.copy", params: { src: "web/static/index.html", dest: "app/web/static/index.html" } },
    { method: "fs.copy", params: { src: "web/static/styles.css", dest: "app/web/static/styles.css" } },
    { method: "fs.copy", params: { src: "web/static/app.js", dest: "app/web/static/app.js" } },
    { method: "fs.copy", params: { src: "web/static/components.js", dest: "app/web/static/components.js" } },
    { method: "fs.copy", params: { src: "web/static/hooks.js", dest: "app/web/static/hooks.js" } },
    { method: "fs.copy", params: { src: "web/static/api.js", dest: "app/web/static/api.js" } },
    { method: "fs.copy", params: { src: "web/static/constants.js", dest: "app/web/static/constants.js" } },
    { method: "fs.copy", params: { src: "web/static/icons.js", dest: "app/web/static/icons.js" } },
    { method: "fs.copy", params: { src: "web/static/Logo_1.png", dest: "app/web/static/Logo_1.png" } },
    { method: "fs.copy", params: { src: "web/static/default.jpg", dest: "app/web/static/default.jpg" } },
    // 9. Apply flash attention fix for Windows compatibility
    { method: "fs.copy", params: { src: "patches/builders.py", dest: "app/codeclm/models/builders.py" } },
    // 10. Apply Demucs 24-bit quality fix (was 16-bit)
    { method: "fs.copy", params: { src: "patches/demucs/apply.py", dest: "app/third_party/demucs/models/apply.py" } },
    // 11. Apply separate mode fix (generate tokens once, decode 3 times for vocal/bgm/mixed)
    { method: "fs.copy", params: { src: "patches/gradio/levo_inference_lowmem.py", dest: "app/tools/gradio/levo_inference_lowmem.py" } }
  ]
}