vyluong commited on
Commit
57d5a31
·
verified ·
1 Parent(s): ee08abf

Update app/core/config.py

Browse files
Files changed (1) hide show
  1. app/core/config.py +3 -2
app/core/config.py CHANGED
@@ -46,9 +46,10 @@ class Settings(BaseSettings):
46
  # pyannote/speaker-diarization-3.1
47
  # pyannote/speaker-diarization-community-1
48
  pyannote_model: str = "pyannote/speaker-diarization-community-1"
49
- sortformer_model = "nvidia/diar_sortformer_4spk-v1"
50
 
51
- diarization_backend = "sortformer"
 
 
52
 
53
  # Device settings
54
  device: Literal["cuda", "cpu", "auto"] = "auto"
 
46
  # pyannote/speaker-diarization-3.1
47
  # pyannote/speaker-diarization-community-1
48
  pyannote_model: str = "pyannote/speaker-diarization-community-1"
 
49
 
50
+ sortformer_model: str = "nvidia/diar_sortformer_4spk-v1"
51
+
52
+ diarization_backend: str = "sortformer"
53
 
54
  # Device settings
55
  device: Literal["cuda", "cpu", "auto"] = "auto"