forked from efforting.tech/stt-server
Log to stderr when model needs to be downloaded
Checks cache first with local_files_only=True; if the model isn't present logs "downloading model ..." to stderr before WhisperModel triggers the actual download. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
@@ -123,8 +123,15 @@ parser.add_argument('--verbose', '-v', action='store_true')
|
||||
args = parser.parse_args()
|
||||
verbose = args.verbose
|
||||
|
||||
log(f'loading faster-whisper {args.model} ({args.device}, {args.compute_type})...')
|
||||
from faster_whisper import WhisperModel
|
||||
from huggingface_hub import snapshot_download
|
||||
|
||||
try:
|
||||
snapshot_download(f'Systran/faster-whisper-{args.model}', local_files_only=True)
|
||||
except Exception:
|
||||
log(f'downloading model {args.model}...', error=True)
|
||||
|
||||
log(f'loading faster-whisper {args.model} ({args.device}, {args.compute_type})...')
|
||||
try:
|
||||
model = WhisperModel(args.model, device=args.device, compute_type=args.compute_type)
|
||||
log(f'model ready on {args.device}')
|
||||
|
||||
Reference in New Issue
Block a user