Skip to content

  • Projects
  • Groups
  • Snippets
  • Help
  • This project
    • Loading...
  • Sign in / Register
T
Tidal
  • Overview
    • Overview
    • Details
    • Activity
    • Cycle Analytics
  • Repository
    • Repository
    • Files
    • Commits
    • Branches
    • Tags
    • Contributors
    • Graph
    • Compare
    • Charts
  • Issues 0
    • Issues 0
    • List
    • Board
    • Labels
    • Milestones
  • Merge Requests 0
    • Merge Requests 0
  • CI / CD
    • CI / CD
    • Pipelines
    • Jobs
    • Schedules
    • Charts
  • Wiki
    • Wiki
  • Snippets
    • Snippets
  • Members
    • Members
  • Collapse sidebar
  • Activity
  • Graph
  • Charts
  • Create a new issue
  • Jobs
  • Commits
  • Issue Boards
  • PLN
  • Tidal
  • Repository

Switch branch/tag
  • Tidal
  • tools
  • foundry
  • research
  • engine2_spike
  • spike.sh
Find file
BlameHistoryPermalink
  • PLN (Algolia)'s avatar
    feat(foundry): engine2 Roformer VRAM spike — fits the 6GB card (py3.12, 3GB peak) · 1a201dfa
    Task #8. Question: is a SOTA Mel-/BS-Band Roformer reachable on the 6GB RTX 2060,
    or are we stuck on demucs? Answer: it fits comfortably — but the wall was never VRAM.
    
    Run 1 (system python 3.14) died at model instantiation: audio-separator's Roformer
    loader carries `Callable | None` type hints that beartype rejects under 3.14
    (BeartypeDecorHintNonpepException on `stft_window_fn`). Peak VRAM at crash: 921 MiB —
    nowhere near the limit. So engine2 is gated on the interpreter, not the card.
    
    Run 2 (python 3.12 venv) succeeded: bs_roformer_ep_317 separated the 164s Loituma
    source in 90s (~0.55× realtime) at a peak of 3051/6144 MiB — half the card, ~3GB
    headroom. torch 2.12.1+cu130, onnxruntime-gpu 1.27, CUDA detected fine.
    
    Caveat for the wiring task (#9): ep_317 is a 2-stem model (Instrumental/Vocals); the
    Foundry contract is 4 stems (drums/bass/other/vocals). #9 must pick a 4-stem Roformer
    checkpoint and re-confirm VRAM — but with 3GB to spare on a 2-stem pass, a 4-stem run
    is plausible. The spike is self-contained + idempotent (research/engine2_spike/spike.sh,
    SPIKE_PY pins the interpreter); result.json carries the numbers.
    PLN (Algolia) authored Jun 23, 2026
    1a201dfa
spike.sh 4.55 KB
EditWeb IDE
×

Replace spike.sh

Attach a file by drag & drop or click to upload


Cancel
A new branch will be created in your fork and a new merge request will be started.