mirrors/whispering
1
0
Fork 0
mirror of https://github.com/shirayu/whispering.git synced 2024-11-21 08:21:02 +00:00
Code Issues Projects Releases Packages Wiki Activity

Updated messages

Browse source
This commit is contained in:
Yuta Hayashibe 2023-01-07 01:18:41 +09:00
parent 1eadec5466
commit 472a02712d
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
whispering/transcriber.py
View file

@ -30,9 +30,9 @@ class WhisperStreamingTranscriber:
self.dtype = torch.float16 if fp16 else torch.float32
if self.model.device == torch.device("cpu"):
if torch.cuda.is_available():
logger.warning("Performing inference on CPU when CUDA is available")
logger.info("Performing inference on CPU though CUDA is available")
if self.dtype == torch.float16:
logger.warning("FP16 is not supported on CPU; using FP32 instead")
logger.info("Using FP32 because FP16 is not supported on CPU")
self.dtype = torch.float32
if self.dtype == torch.float32: