From 101e913f849ce450dea400a4681095d6c39d455f Mon Sep 17 00:00:00 2001 From: "Schmieder, Jacob" Date: Thu, 10 Oct 2024 09:29:48 +0000 Subject: [PATCH] make ruff happy --- scraibe/diarisation.py | 2 +- scraibe/transcriber.py | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/scraibe/diarisation.py b/scraibe/diarisation.py index 6e6d6b9..eeef135 100644 --- a/scraibe/diarisation.py +++ b/scraibe/diarisation.py @@ -37,7 +37,7 @@ from pyannote.audio import Pipeline from pyannote.audio.pipelines.speaker_diarization import SpeakerDiarization from torch import Tensor from torch import device as torch_device -from torch.cuda import is_available + from huggingface_hub import HfApi from huggingface_hub.utils import RepositoryNotFoundError diff --git a/scraibe/transcriber.py b/scraibe/transcriber.py index 9c891f6..040b79d 100644 --- a/scraibe/transcriber.py +++ b/scraibe/transcriber.py @@ -31,7 +31,6 @@ from faster_whisper import WhisperModel as FasterWhisperModel from faster_whisper.tokenizer import _LANGUAGE_CODES as FASTER_WHISPER_LANGUAGE_CODES from typing import TypeVar, Union, Optional from torch import Tensor, device -from torch.cuda import is_available as cuda_is_available from numpy import ndarray from inspect import signature from abc import abstractmethod