Skip to main content

pymss-core

中文文档

Core model, configuration, and checkpoint package for music source separation.

pymss-core is the shared low-level package for higher-level projects such as pymss inference and pymsst training. It contains model definitions, configuration loading, and checkpoint compatibility helpers. It intentionally does not include inference DSP pipelines, chunked demixing, audio file I/O, model downloads, catalog management, CLI, HTTP server, WebUI, datasets, losses, or training loops.

Install

pip install pymss-core

For local development:

uv sync --dev

Optional MLX backend on Apple Silicon:

pip install "pymss-core[mlx]"

Public API

from pymss_core import (
    get_model_from_config,
    load_config,
    load_model_weights,
)

model, config = get_model_from_config("bs_roformer", "config.yaml")
load_model_weights(model, "model.ckpt", model_type="bs_roformer", strict=True)

model.eval()

Package Boundary

Included:

  • YAML config loading with AttrDict
  • PyTorch model definitions under pymss_core.modules
  • Optional MLX backend implementations for supported model forward paths
  • Model factory: get_model_from_config(model_type, config_path)
  • Checkpoint helpers for common MSS checkpoint containers
  • Small model-internal DSP math needed to construct model structures
  • VR network structures and VR model parameter JSON files

Excluded:

  • Audio file decoding/encoding
  • Resampling, preprocessing, and full inference DSP pipelines
  • Tensor-level chunked demixing runtime
  • Model catalog, aliases, downloads, and cache management
  • CLI, server, WebUI, and endpoint schemas
  • Dataset, augmentation, loss, metrics, and trainer code
  • Any default dependency on MLX, Librosa, tqdm, Lightning, FastAPI, Uvicorn, PyAV, WandB, or training extras

Repository Roles

pymss-core
  shared model/config/checkpoint layer

pymss
  user-facing inference package built on pymss-core, with audio I/O and demix

pymsst
  training package built on pymss-core, with training data/loss/runtime code

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

pymss_core-0.1.6.tar.gz (104.3 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

pymss_core-0.1.6-py3-none-any.whl (137.1 kB view details)

Uploaded Python 3

File details

Details for the file pymss_core-0.1.6.tar.gz.

File metadata

  • Download URL: pymss_core-0.1.6.tar.gz
  • Upload date:
  • Size: 104.3 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.14

File hashes

Hashes for pymss_core-0.1.6.tar.gz
Algorithm Hash digest
SHA256 c3eb5f6913af1add8b51bb02cf218bc604568a52bd446bbe95214dd1dc14c5af
MD5 35665f6d87900539621f8b21c1197fa6
BLAKE2b-256 287303e7656ba1835c3a6f16c14669994feaba292639189f6fed256e55e7a259

See more details on using hashes here.

Provenance

The following attestation bundles were made for pymss_core-0.1.6.tar.gz:

Publisher: release.yml on pymss-project/pymss-core

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file pymss_core-0.1.6-py3-none-any.whl.

File metadata

  • Download URL: pymss_core-0.1.6-py3-none-any.whl
  • Upload date:
  • Size: 137.1 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.14

File hashes

Hashes for pymss_core-0.1.6-py3-none-any.whl
Algorithm Hash digest
SHA256 528bd36d21e477083f4770b44fe275654d69f752069d97d330ca55f0ef013788
MD5 0f6f47b44aa89963af177aa443551903
BLAKE2b-256 5fb11f2062395bc75a8cd4e102ee318bdd513c876f54a352a13ca42efeef6f4e

See more details on using hashes here.

Provenance

The following attestation bundles were made for pymss_core-0.1.6-py3-none-any.whl:

Publisher: release.yml on pymss-project/pymss-core

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page