benkaron commited on
Commit
3ddc213
·
verified ·
1 Parent(s): 77c7af9

BS-RoFormer wind stem for slurper, fp16 Core ML with golden chunk

Browse files
.gitattributes CHANGED
@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ golden_horns.f32 filter=lfs diff=lfs merge=lfs -text
37
+ golden_raw.f32 filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: coreml
3
+ tags:
4
+ - audio
5
+ - music-source-separation
6
+ - bs-roformer
7
+ - coreml
8
+ base_model: noblebarkrr/BS-Roformer-MVSep-Mega-53-stems
9
+ ---
10
+
11
+ # BSRoformer-Wind-CoreML
12
+
13
+ The wind stem (brass and woodwinds) of the [MVSep Mega 53-stem BS-RoFormer](https://github.com/ZFTurbo/Music-Source-Separation-Training/releases/tag/v1.0.21), converted to Core ML for [slurper](https://github.com/trevors/slurper), where it produces the "horns" stem.
14
+
15
+ - `bsr_wind_fp16.mlpackage`: the graph `frames[1,2,690,2048] -> recon[1,2,690,2048]` for one 8 s chunk at 44.1 kHz, framed with a 2048-sample periodic Hann window at hop 512: windowed DFT, band split, the axial rotary transformer, mask estimator, complex mask multiply and inverse DFT, all inside the model in float16. The host reflect-pads and frames the chunk and overlap-adds the output divided by the summed squared window.
16
+ - `golden_raw.f32`, `golden_horns.f32`: an 8 s stereo chunk (channel-major float32) and PyTorch's output for it, used by slurper's tests. Core ML matches PyTorch on it at cosine 0.99998 (44.7 dB SDR).
17
+
18
+ Produced by `scripts/convert_bs_roformer.py` in the slurper repository from the single-stem repack [noblebarkrr/BS-Roformer-MVSep-Mega-53-stems](https://huggingface.co/noblebarkrr/BS-Roformer-MVSep-Mega-53-stems) (`v1/bs_mega_53stem_wind_mvsep.ckpt`), with the model code from [Music-Source-Separation-Training](https://github.com/ZFTurbo/Music-Source-Separation-Training) (MIT). See the upstream release for the weights' terms.
19
+
20
+ The golden chunk is 8 s from 0:32 of "Blues for Mundy" by The Airmen of Note, The United States Air Force Band (composed by Rick Whitehead), from *60 Years of the Airmen of Note* (2011), via [Wikimedia Commons](https://commons.wikimedia.org/wiki/File:Blues_for_Mundy_-_Airmen_of_Note_-_United_States_Air_Force_Band.mp3). As a work of the US federal government the recording is in the public domain.
bsr_wind_fp16.mlpackage/Data/com.apple.CoreML/model.mlmodel ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dbb26610e43376763231deaf8a52f3e296ce50213dd6bb8ac1a452c6f2c70e1e
3
+ size 748990
bsr_wind_fp16.mlpackage/Data/com.apple.CoreML/weights/weight.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:acc4dc39bc044f547da86ac399300fdf41f9fed5c114e901a0766b7c5e36fb32
3
+ size 94421760
bsr_wind_fp16.mlpackage/Manifest.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "fileFormatVersion": "1.0.0",
3
+ "itemInfoEntries": {
4
+ "B0268636-0C8F-41A3-9A12-8E5E4EB52A7F": {
5
+ "author": "com.apple.CoreML",
6
+ "description": "CoreML Model Weights",
7
+ "name": "weights",
8
+ "path": "com.apple.CoreML/weights"
9
+ },
10
+ "F5730EEA-E431-4FBB-B005-B973B8041A83": {
11
+ "author": "com.apple.CoreML",
12
+ "description": "CoreML Model Specification",
13
+ "name": "model.mlmodel",
14
+ "path": "com.apple.CoreML/model.mlmodel"
15
+ }
16
+ },
17
+ "rootModelIdentifier": "F5730EEA-E431-4FBB-B005-B973B8041A83"
18
+ }
golden_horns.f32 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0f5ed9182b7d87deaabaa778cbbe7a42e68ac0d1379b0bcad50b436f88973eef
3
+ size 2822400
golden_raw.f32 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4e91dd446ccc25ce3c597d4aa8dee7f5e40ee76afcb08e56d0a07bca750ca37f
3
+ size 2822400