Add CI workflow for testing and update model registration

Author: lifeiteng

.github/workflows/test.yml

@@ -0,0 +1,45 @@
+name: Run Tests
+
+on:
+  push:
+    branches: [ main, dev, develop ]
+  pull_request:
+    branches: [ main, dev, develop ]
+  workflow_dispatch:
+
+jobs:
+  test:
+    name: Test on ${{ matrix.os }} with Python ${{ matrix.python-version }}
+    runs-on: ${{ matrix.os }}
+    strategy:
+      fail-fast: false
+      matrix:
+        os: [ubuntu-latest, macos-latest]
+        python-version: ['3.10', '3.11', '3.12']
+
+    steps:
+      - uses: actions/checkout@v4
+        with:
+          submodules: recursive
+
+      - name: Set up Python ${{ matrix.python-version }}
+        uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python-version }}
+
+      - name: Cache pip dependencies
+        uses: actions/cache@v4
+        with:
+          path: ~/.cache/pip
+          key: ${{ runner.os }}-pip-${{ hashFiles('setup.py') }}
+          restore-keys: |
+            ${{ runner.os }}-pip-
+
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip setuptools wheel
+          pip install -e .[test]
+
+      - name: Run tests with pytest
+        run: |
+          python tests/model_test.py

setup.py

@@ -20,11 +20,13 @@
     readme="README.md",
     python_requires=">=3.8",
     install_requires=[
+        "kaldi-native-fbank",
         "modelscope>=1.18.0",
         "funasr>=1.1.6",
         "lhotse>=1.24.2",
         "kaldialign",
         "torch",
+        "torchaudio",
     ],
     extras_require={
         "test": [
@@ -42,6 +44,7 @@
         "Programming Language :: Python :: 3.11",
         "Programming Language :: Python :: 3.12",
         "Programming Language :: Python :: 3.13",
+        "Programming Language :: Python :: 3.14",
         "Intended Audience :: Science/Research",
         "Operating System :: POSIX :: Linux",
         "Operating System :: MacOS :: MacOS X",

src/omnisense/models/model.py

@@ -562,7 +562,7 @@ def forward(
         return xs_pad, olens
 
 
-@tables.register("model_classes", "SenseVoiceSmall")
+@tables.register("model_classes", "iic/SenseVoiceSmall")
 class SenseVoiceSmall(nn.Module):
     """CTC-attention hybrid Encoder-Decoder model"""