administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-05-01 19:17:41 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

fix linter

Browse Source
This commit is contained in:
Josh Hawkins 2025-05-26 07:49:27 -05:00
parent fa730441bf
commit e60b51d036
1 changed files with 3 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
frigate/data_processing/real_time/whisper_online.py
View File

@ -119,15 +119,6 @@ class FasterWhisperASR(ASRBase):
from faster_whisper import WhisperModel
logging.getLogger("faster_whisper").setLevel(logging.WARNING)
if model_dir is not None:
logger.debug(
f"Loading whisper model from model_dir {model_dir}. modelsize and cache_dir parameters are not used."
)
model_size_or_path = model_dir
elif modelsize is not None:
model_size_or_path = modelsize
else:
raise ValueError("modelsize or model_dir parameter must be set")
# this worked fast and reliably on NVIDIA L40
model = WhisperModel(
@ -514,10 +505,10 @@ class OnlineASRProcessor:
p = self.commited[:k]
p = [t for _, _, t in p]
prompt = []
l = 0
while p and l < 200: # 200 characters prompt size
y = 0
while p and y < 200: # 200 characters prompt size
x = p.pop(-1)
l += len(x) + 1
y += len(x) + 1
prompt.append(x)
non_prompt = self.commited[k:]
return self.asr.sep.join(prompt[::-1]), self.asr.sep.join(