fish-speech/docs/es/inference.md

Inferencia

El modelo Fish Audio S2 requiere una gran cantidad de VRAM. Recomendamos usar una GPU con al menos 24GB para la inferencia.

Descargar Pesos

Primero, necesitas descargar los pesos del modelo:

hf download fishaudio/s2-pro --local-dir checkpoints/s2-pro

Inferencia por Línea de Comandos

!!! note Si planeas dejar que el modelo elija aleatoriamente el timbre de voz, puedes omitir este paso.

1. Obtener tokens VQ a partir de audio de referencia

python fish_speech/models/dac/inference.py \
    -i "test.wav" \
    --checkpoint-path "checkpoints/s2-pro/codec.pth"

Deberías obtener un fake.npy y un fake.wav.

2. Generar tokens semánticos a partir de texto:

python fish_speech/models/text2semantic/inference.py \
    --text "El texto que quieres convertir" \
    --prompt-text "Tu texto de referencia" \
    --prompt-tokens "fake.npy" \
    # --compile

Este comando creará un archivo codes_N en el directorio de trabajo, donde N es un entero que comienza desde 0.

!!! note Puede que quieras usar --compile para fusionar kernels CUDA y acelerar la inferencia. Sin embargo, recomendamos usar nuestra optimización de aceleración de inferencia con sglang. Correspondientemente, si no planeas usar aceleración, puedes comentar el parámetro --compile.

!!! info Para GPUs que no soportan bf16, puede que necesites usar el parámetro --half.

3. Generar audio a partir de tokens semánticos:

python fish_speech/models/dac/inference.py \
    -i "codes_0.npy" \

Después de eso, obtendrás un archivo fake.wav.

Inferencia con WebUI

Próximamente.