Upload config.yaml with huggingface_hub

config.yaml

@@ -0,0 +1,51 @@
+# NeuroRVQ safetensors weights for neurorvq-rs
+# This file enables HuggingFace download tracking
+# https://huggingface.co/docs/hub/models-download-stats
+
+model_type: neurorvq
+framework: burn
+language: rust
+
+modalities:
+  - EEG
+  - ECG
+  - EMG
+
+tokenizers:
+  - NeuroRVQ_EEG_tokenizer_v1.safetensors
+  - NeuroRVQ_ECG_tokenizer_v1.safetensors
+  - NeuroRVQ_EMG_tokenizer_v1.safetensors
+
+foundation_models:
+  - NeuroRVQ_EEG_foundation_model_v1.safetensors
+  - NeuroRVQ_EMG_foundation_model_v1.safetensors
+
+architecture:
+  encoder_depth: 12
+  decoder_depth: 3
+  num_heads: 10
+  codebook_size: 8192
+  code_dim: 128
+  eeg:
+    embed_dim: 200
+    patch_size: 200
+    n_patches: 256
+    n_channels: 103
+    rvq_levels: 8
+    params_tokenizer: 76.0M
+    params_fm: 58.6M
+  ecg:
+    embed_dim: 40
+    patch_size: 40
+    n_patches: 600
+    n_channels: 15
+    rvq_levels: 8
+    params_tokenizer: 68.1M
+  emg:
+    embed_dim: 200
+    patch_size: 200
+    n_patches: 256
+    n_channels: 16
+    rvq_levels: 16
+    params_tokenizer: 143.6M
+    params_fm: 111.2M