Transcribix-Pinokio/install.js at main · C0m3b4ck/Transcribix-Pinokio · 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
92
93
94
95
module.exports = {
  requires: {
    bundle: "ai"
  },
  run: [
    {
      method: "local.set",
      params: {
        state: {
          group: "{{args.group}}"
        }
      }
    },
    {
      when: "{{gpu === 'nvidia' && platform === 'linux'}}",
      method: "shell.run",
      params: {
        message: [
          "sudo apt install -y libcublas12",
        ],
      }
    },
    // Whisper Variants
    {
      when: "{{args.group === 'whisper'}}",
      method: "shell.run",
      params: {
        venv: "venvs/whisper",
        path: ".",
        message: [
          "uv pip install -r requirements/whisper.txt",
          "uv pip install -r requirements/gradio.txt",
        ],
      }
    },
    // NVIDIA Models
    {
      when: "{{args.group === 'nvidia'}}",
      method: "shell.run",
      params: {
        venv: "venvs/nvidia",
        path: ".",
        message: [
          "uv pip install -r requirements/nvidia.txt",
          "uv pip install -r requirements/gradio.txt",
        ],
      }
    },
    // Lightweight Models
    {
      when: "{{args.group === 'lightweight'}}",
      method: "shell.run",
      params: {
        venv: "venvs/lightweight",
        path: ".",
        message: [
          "uv pip install -r requirements/lightweight.txt",
          "uv pip install -r requirements/gradio.txt",
        ],
      }
    },
    // All Models
    {
      when: "{{args.group === 'all'}}",
      method: "shell.run",
      params: {
        venv: "venvs/all",
        path: ".",
        message: [
          "uv pip install -r requirements/all.txt",
          "uv pip install -r requirements/gradio.txt",
        ],
      }
    },
    // Install torch for the selected group
    {
      method: "script.start",
      params: {
        uri: "torch.js",
        params: {
          venv: "venvs/{{args.group}}",
          path: "."
        }
      }
    },
    // Post-install message
    {
      method: "input",
      params: {
        title: "Install Complete",
        description: "Installation complete. You may now start the app."
      }
    },
  ]
}