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()
|
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}')
|
||||||
|
|||||||
Reference in New Issue
Block a user