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:
2026-06-07 09:07:13 +00:00
parent 6bbc04dde7
commit 218687b039

View File

@@ -123,8 +123,15 @@ parser.add_argument('--verbose', '-v', action='store_true')
args = parser.parse_args() args = parser.parse_args()
verbose = args.verbose verbose = args.verbose
log(f'loading faster-whisper {args.model} ({args.device}, {args.compute_type})...')
from faster_whisper import WhisperModel 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: try:
model = WhisperModel(args.model, device=args.device, compute_type=args.compute_type) model = WhisperModel(args.model, device=args.device, compute_type=args.compute_type)
log(f'model ready on {args.device}') log(f'model ready on {args.device}')