e2cd40e2bfc74aef8c2e3f96e6b0b70cb160f10b
- Added global model caching with FIFO eviction policy. - Added support for float16 and bfloat16 precision. - Updated model list and names for Demucs v4 compatibility. - Integrated ComfyUI native progress bar with robust callback. - Updated README and node metadata with emojis (Bolt style). - Fixed potential OOM and attribute errors. - Updated requirements.txt with version constraints. Co-authored-by: Meisoftcoltd <38667978+Meisoftcoltd@users.noreply.github.com>
ComfyUI-Demucs-Pro
Nodo personalizado para ComfyUI que permite la separación de fuentes de audio utilizando la tecnología Meta Demucs (v4). Ideal para extraer voces, baterías, bajos y otros instrumentos con alta fidelidad.
Características
- Soporte Demucs v4: Incluye modelos Hybrid Transformer (HTDemucs).
- Separación de hasta 6 pistas: Voces, Batería, Bajo, Otros, y soporte para Guitarra y Piano (en modelos compatibles).
- Optimizado para NVIDIA RTX 3090: Aprovecha la aceleración CUDA y soporte para precisiones optimizadas (bfloat16/float16).
- Caché Global de Modelos: Los modelos se mantienen en memoria para un intercambio ultrarrápido.
- Barra de Progreso Nativa: Integración completa con la barra de progreso de ComfyUI.
- Descarga Automática: Los modelos se descargan automáticamente a
ComfyUI/models/demucs. - Estabilidad Mejorada: Opción
splitpara evitar errores de memoria (OOM) en audios largos.
Instalación
Vía ComfyUI Manager (Recomendado)
- Busca
ComfyUI-Demucs-Proen el Manager. - Haz clic en Install.
- Reinicia ComfyUI.
Instalación Manual
- Clona este repositorio en tu carpeta
custom_nodes:cd ComfyUI/custom_nodes git clone https://github.com/usuario/ComfyUI-Demucs-Pro - Instala las dependencias:
pip install -r requirements.txt - Reinicia ComfyUI.
Parámetros del Nodo
- audio: El clip de audio a procesar.
- model:
htdemucs: Modelo estándar de 4 pistas.htdemucs_ft: Versión fine-tuned del modelo estándar.htdemucs_6s: Modelo de 6 pistas (incluye Guitarra y Piano).hdemucs_mmi: Basado en HDemucs con mejoras.mdx/mdx_extra: Modelos basados en Music Demixing Challenge.
- device:
cudapara usar la GPU ocpu. - precision:
float32,bfloat16,float16.bfloat16es recomendado para arquitecturas Ampere (RTX 3090/4090). - shifts: Número de predicciones aleatorias (mayor valor = mayor calidad, pero más lento). Recomendado: 1-5.
- overlap: Solapamiento entre ventanas de procesamiento. Por defecto: 0.25.
- split: Divide el audio en segmentos para ahorrar VRAM. Imprescindible para tarjetas con menos de 8GB o audios largos.
- stems: Activa o desactiva las salidas específicas para cada instrumento.
Recomendaciones de Hardware
Para un rendimiento óptimo, se recomienda el uso de GPUs de gama alta:
- NVIDIA RTX 3090 / 4090: Rendimiento excepcional, permitiendo procesar canciones completas en segundos con altos valores de
shifts. - VRAM: Se recomienda al menos 8GB para modelos
htdemucs. Si encuentras errores de memoria, asegúrate de tener activada la opciónsplity usarbfloat16.
Reporte de Errores
Si encuentras algún problema, por favor abre un Issue en el repositorio de GitHub detallando tu sistema operativo, modelo de GPU y el error obtenido.
Desarrollado para la comunidad de ComfyUI por un Senior Python Developer especializado en IA Multimedia (⚡ Bolt Edition).
Languages
Python
100%