smank commited on
Commit
b1da294
·
verified ·
1 Parent(s): 469b019

Model card: family README with htdemucs_ft, license + attribution

Browse files
Files changed (1) hide show
  1. README.md +38 -3
README.md CHANGED
@@ -1,3 +1,38 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags:
4
+ - audio
5
+ - music-source-separation
6
+ - stem-separation
7
+ - onnx
8
+ - demucs
9
+ ---
10
+
11
+ # HTDemucs — ONNX conversions
12
+
13
+ ONNX conversions of the Demucs v4 (Hybrid Transformer) family, packaged for
14
+ [SonicDemixer](https://smank.io) (on-device stem separation for macOS). All
15
+ graphs share one contract: input `mix` `[1, 2, segmentSamples]` float32
16
+ planar waveform → output `sources` `[1, numStems, 2, segmentSamples]`,
17
+ opset 17, STFT/iSTFT embedded in the graph (no complex dtypes).
18
+
19
+ | File | Stems | Notes |
20
+ |---|---|---|
21
+ | `htdemucs.onnx` | drums, bass, other, vocals | baseline 4-stem |
22
+ | `htdemucs_6s.onnx` | + guitar, piano | 6-source variant |
23
+ | `htdemucs_ft.onnx` | drums, bass, other, vocals | the fine-tuned bag's four per-source models fused into ONE graph — each submodel contributes its specialty stem (mapping verified against the bag weights). Higher quality, ~4× the compute. |
24
+
25
+ Integrity: files are LFS-tracked; the LFS OID is the SHA-256 that consuming
26
+ apps verify after download.
27
+
28
+ ## License & attribution
29
+
30
+ MIT. These are derivative conversions of:
31
+
32
+ - **Demucs / HTDemucs** — code and pretrained weights © Meta Platforms (FAIR),
33
+ MIT. https://github.com/facebookresearch/demucs
34
+ - **ONNX-exportable HTDemucs + conv-based STFT/iSTFT** — Mixxx Development
35
+ Team demucs fork (GSoC 2025, Anmol Mishra), MIT.
36
+ https://github.com/mixxxdj/demucs
37
+
38
+ Conversion scripts: `tools/model-conversion/` in the SonicDemixer repository.