ACE-Step-CPU / train_engine.py

Commit History

  1. fix understand_audio: clone tensors for inference mode, working on GPU (52s)

    4b2f4ad
  2. add understand_audio (LM reverse), demucs-infer fix, commit refs, dtype fixes

    6bfdc38
  3. major update: PyTorch inference, Gradio 6, session isolation, /understand captioning⁠

    ff239f5

  1. SDPA first on Blackwell, FA2 only for Ampere/Hopper, txt caption support

    04ccf32
  2. add GPU/CUDA auto-detect, mixed precision, flash_attn, txt caption parser

    917e4ed
  3. save only on cancel/finish, max epochs 1000, default 3

    d42aa91
  4. add mid/sas analysis modes (Demucs + ensemble), auto-select by dataset size

    b38d0b1
  5. add auto-captioning (BPM/key/signature via librosa), add librosa+mutagen deps

    1d42836
  6. add XL variant mappings to _VARIANT_DIR

    3c5ce9c
  7. fix: adapter saved to clean dir, LM dropdown no 'Default', on-demand download

    e62602f
  8. Side-Step training engine, tested locally on CPU⁠

    a07b39d