Feature/mac mps support#23

Open

TheApeMachine wants to merge 5 commits intoHeartMuLa:mainfrom

TheApeMachine:feature/mac-mps-support

Commits on Jan 19, 2026

Refactor music generation pipeline to support dynamic device and dtype selection. Update argument handling in `run_music_generation.py` and improve `HeartMuLaGenPipeline` class for better input pro…
TheApeMachine
committed
Enhance device and dtype handling in lyrics transcription and heart codec model. Update `run_lyrics_transcription.py` to dynamically select device based on availability, and modify `HeartCodec` to …
TheApeMachine
committed
Refactor `HeartMuLaGenPipeline` to improve autocast handling and optimize audio token padding. Introduce a context manager for autocast that gracefully handles unsupported cases, and preallocate bu…
TheApeMachine
committed
Implement Metal support in `torchtune_metal.py` for optimized inference on MPS. Update `pyproject.toml` to include the optimizer package directory. Enhance `HeartMuLaGenPipeline` to optionally enab…
TheApeMachine
committed
Implement Metal support for RMSNorm and RoPE operations, including new Metal kernels and Python wrappers. Update `pyproject.toml` to remove the optimizer package directory. Enhance runtime detectio…
TheApeMachine
committed