Merge pull request #477 from niwa2/hf_token_via_env

add option to provide HuggingFace-token via environment variable
jhj0517 · Jan 24, 2025 · e44f57f · e44f57f
2 parents 77b3814 + 6ab697e
commit e44f57f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modules/whisper/base_transcription_pipeline.py b/modules/whisper/base_transcription_pipeline.py
@@ -179,7 +179,7 @@ def run(self,
         if diarization_params.is_diarize:
             result, elapsed_time_diarization = self.diarizer.run(
                 audio=origin_audio,
-                use_auth_token=diarization_params.hf_token,
+                use_auth_token=diarization_params.hf_token if diarization_params.hf_token else os.environ.get("HF_TOKEN"),
                 transcribed_result=result,
                 device=diarization_params.diarization_device
             )