selfhostyourtech/reflector
2
0
Fork 0
mirror of https://github.com/Monadical-SAS/reflector.git synced 2025-12-20 20:29:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Move HF_token to modal secret (#354)

Browse Source 
* update all modal deployments and change seamless configuration due to change in src repo

* add fixture

* move token to secret
This commit is contained in:
projects-g
2024-04-19 10:30:45 +05:30
committed by GitHub
parent 72b22d1005
commit 63502becd6
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
server/gpu/modal/reflector_diarizer.py
View File

@@ -11,7 +11,7 @@ from pydantic import BaseModel
PYANNOTE_MODEL_NAME: str = "pyannote/speaker-diarization-3.0" PYANNOTE_MODEL_NAME: str = "pyannote/speaker-diarization-3.0"
MODEL_DIR = "/root/diarization_models" MODEL_DIR = "/root/diarization_models"
HUGGINGFACE_TOKEN = modal.Secret.from_name("my-huggingface-secret")
stub = Stub(name="reflector-diarizer") stub = Stub(name="reflector-diarizer")
@@ -34,7 +34,7 @@ def download_pyannote_audio():
Pipeline.from_pretrained( Pipeline.from_pretrained(
"pyannote/speaker-diarization-3.0", "pyannote/speaker-diarization-3.0",
cache_dir=MODEL_DIR, cache_dir=MODEL_DIR,
use_auth_token="***REMOVED***" use_auth_token=HUGGINGFACE_TOKEN
) )