fix(transcriber): распознавание cublas/cudnn ошибок и уточнение версии CUDA
- Зачем: - cublas64_12.dll not found не распознавалась как CUDA-ошибка, fallback на CPU не срабатывал - Что: - _is_cuda_error теперь матчит cublas и cudnn в дополнение к cuda и out of memory - README: добавлен шаг GPU-ускорения в установку, зафиксирована CUDA 12 (ctranslate2 4.7) - Проверка: - uv run pytest (97 passed) Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
@@ -225,7 +225,7 @@ def _create_model(model_name: str, device: str, compute_type: str):
|
||||
|
||||
def _is_cuda_error(exc: BaseException) -> bool:
|
||||
msg = str(exc).lower()
|
||||
return "cuda" in msg or "out of memory" in msg
|
||||
return any(k in msg for k in ("cuda", "cublas", "cudnn", "out of memory"))
|
||||
|
||||
|
||||
def _is_missing_socksio_error(exc: BaseException) -> bool:
|
||||
|
||||
Reference in New Issue
Block a user