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.5.tar.gz (101.8 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.5-py3-none-any.whl (133.0 kB view details)

Uploaded Python 3

File details

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

File metadata

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

File hashes

Hashes for pymss_core-0.1.5.tar.gz
Algorithm Hash digest
SHA256 7d546cab1be3f02b06caba1fafb88c092d51b3fc08810489ad44d1b8edc721eb
MD5 06141cf06afa54eb2f0863dc811deaf2
BLAKE2b-256 740d6558bfbf2f77cb91df60f890aece077f5e6743385893ed4413d2b898bcbc

See more details on using hashes here.

Provenance

The following attestation bundles were made for pymss_core-0.1.5.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.5-py3-none-any.whl.

File metadata

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

File hashes

Hashes for pymss_core-0.1.5-py3-none-any.whl
Algorithm Hash digest
SHA256 4226f990b273c30342ad2a48f2bd3bc6197b51798d54e6516074689e66306608
MD5 8f8f4590c2cb1320fbbfd81bd50c931f
BLAKE2b-256 4b12dd8519410670315f2c58066ab73e2f77232114bd069257aac132df145fa9

See more details on using hashes here.

Provenance

The following attestation bundles were made for pymss_core-0.1.5-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