mirror of
https://github.com/Monadical-SAS/reflector.git
synced 2026-04-25 22:55:18 +00:00
fix: add parakeet as default transcriber and fix diarizer image (#939)
This commit is contained in:
committed by
GitHub
parent
61d6fbd344
commit
cc9c5cd4a5
@@ -114,8 +114,8 @@ modal secret create reflector-gpu REFLECTOR_GPU_APIKEY="$API_KEY"
|
|||||||
|
|
||||||
# --- Deploy Functions ---
|
# --- Deploy Functions ---
|
||||||
echo ""
|
echo ""
|
||||||
echo "Deploying transcriber (Whisper)..."
|
echo "Deploying transcriber (Parakeet)..."
|
||||||
TRANSCRIBER_URL=$(modal deploy reflector_transcriber.py 2>&1 | grep -o 'https://[^ ]*web.modal.run' | head -1)
|
TRANSCRIBER_URL=$(modal deploy reflector_transcriber_parakeet.py 2>&1 | grep -o 'https://[^ ]*web.modal.run' | head -1)
|
||||||
if [ -z "$TRANSCRIBER_URL" ]; then
|
if [ -z "$TRANSCRIBER_URL" ]; then
|
||||||
echo "Error: Failed to deploy transcriber. Check Modal dashboard for details."
|
echo "Error: Failed to deploy transcriber. Check Modal dashboard for details."
|
||||||
exit 1
|
exit 1
|
||||||
|
|||||||
@@ -113,12 +113,14 @@ def download_pyannote_audio():
|
|||||||
|
|
||||||
|
|
||||||
diarizer_image = (
|
diarizer_image = (
|
||||||
modal.Image.debian_slim(python_version="3.10")
|
modal.Image.from_registry(
|
||||||
|
"nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04", add_python="3.10"
|
||||||
|
)
|
||||||
.pip_install(
|
.pip_install(
|
||||||
"pyannote.audio==3.1.0",
|
"pyannote.audio==3.1.0",
|
||||||
"requests",
|
"requests",
|
||||||
"onnx",
|
"onnx",
|
||||||
"torchaudio",
|
"torchaudio==2.0.1",
|
||||||
"onnxruntime-gpu",
|
"onnxruntime-gpu",
|
||||||
"torch==2.0.0",
|
"torch==2.0.0",
|
||||||
"transformers==4.34.0",
|
"transformers==4.34.0",
|
||||||
@@ -133,14 +135,6 @@ diarizer_image = (
|
|||||||
secrets=[modal.Secret.from_name("hf_token")],
|
secrets=[modal.Secret.from_name("hf_token")],
|
||||||
)
|
)
|
||||||
.run_function(migrate_cache_llm)
|
.run_function(migrate_cache_llm)
|
||||||
.env(
|
|
||||||
{
|
|
||||||
"LD_LIBRARY_PATH": (
|
|
||||||
"/usr/local/lib/python3.10/site-packages/nvidia/cudnn/lib/:"
|
|
||||||
"/opt/conda/lib/python3.10/site-packages/nvidia/cublas/lib/"
|
|
||||||
)
|
|
||||||
}
|
|
||||||
)
|
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user