Skip to content

Fix MPS out of memory error by disabling PyTorch memory watermark #10

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
lmangani merged 2 commits into from
Jan 29, 2026
Merged

Fix MPS out of memory error by disabling PyTorch memory watermark #10

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
a482160
Initial plan
Copilot Jan 29, 2026
25d90b6
Add PYTORCH_MPS_HIGH_WATERMARK_RATIO=0.0 to fix MPS memory allocation…
Copilot Jan 29, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions backend/app/services/music_service.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1858,6 +1858,7 @@ def _pipeline_callback(progress, msg):

def _run_pipeline():
os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
os.environ["PYTORCH_MPS_HIGH_WATERMARK_RATIO"] = "0.0"

sound_tags = request.tags if request.tags and request.tags.strip() else "pop music"

Expand Down
1 change: 1 addition & 0 deletions launcher.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -55,6 +55,7 @@ def setup_environment():
os.environ["HEARTMULA_GENERATED_AUDIO_DIR"] = str(generated_audio_dir)
os.environ["HEARTMULA_REF_AUDIO_DIR"] = str(ref_audio_dir)
os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1" # Enable Metal Performance Shaders
os.environ["PYTORCH_MPS_HIGH_WATERMARK_RATIO"] = "0.0" # Disable MPS memory limit to prevent OOM errors

# Change working directory to app bundle for frontend serving
# Backend will use absolute paths from environment variables
Expand Down