# 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