Push model using huggingface_hub.

README.md

@@ -1,20 +1,16 @@
 ---
 license: mit
-library_name: softchart
-tags: [taiko, beat-tracking, downbeat, music]
+tags:
+- audio-to-symbolic
+- chart-generation
+- model_hub_mixin
+- music
+- pytorch_model_hub_mixin
+- rhythm-game
+- taiko
 ---
 
-# SoftChart Beat
-
-An 8.1M encoder–decoder with a beat/downbeat head, trained on TJA measure grids.
-On held-out songs: **beat F 0.87, downbeat F 0.62** (@70 ms). Used by SoftChart to
-refine BPM and anchor TJA barlines to detected downbeats (handles pickups).
-
-```python
-from softchart.generate import load_hf
-beat_model = load_hf("JacobLinCool/softchart-beat", device="cuda")
-# beat_model.beat(encoder_output) -> (T/4, 2) beat/downbeat logits
-```
-
-Part of [SoftChart](https://github.com/JacobLinCool/SoftChart). MIT-licensed; trained
-on community-made charts.
+This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
+- Code: https://github.com/JacobLinCool/SoftChart
+- Paper: [More Information Needed]
+- Docs: [More Information Needed]

config.json

@@ -1,15 +1,17 @@
 {
   "aux": true,
-  "beat": true,
+  "beat": "hires",
   "capabilities": {
     "align": false,
     "aux": true,
     "beat_head": true,
-    "ctx": true,
+    "ctx": false,
+    "dual": false,
     "func_time": false,
     "mask_infill": false,
     "plan": false,
-    "sibling": true,
+    "sibling": false,
+    "slot": false,
     "style": false,
     "sync_token": false
   },
@@ -19,6 +21,7 @@
   "ffn": 1024,
   "func_time": false,
   "global_ctx": false,
+  "in_ch": 128,
   "ptr": false,
-  "vocab_size": 1802
+  "vocab_size": 1810
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2b7b97c29ac086a44f674c4c06b4d83bd6932493505cc073340841ca9ef5233c
-size 32539324
+oid sha256:f46e33d74b5511e074e02f4ad63611bd15a2bb2d26770aefb4a41761814d8d2c
+size 33073524