leon-ai/leon
1
1
Fork 0
mirror of https://github.com/leon-ai/leon.git synced 2024-11-30 19:07:39 +03:00
Code Issues Projects Releases Wiki Activity

refactor(python tcp server): ASR CUDA log notification on bootup

Browse Source
This commit is contained in:
louistiti 2024-06-22 21:03:27 +08:00
parent aa63a6dabc
commit 76b197baeb
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tcp_server/src/lib/asr/api.py
View File

@ -22,11 +22,9 @@ class ASR:
self.log('Loading model...') self.log('Loading model...')
if device == 'auto': if device == 'auto':
if torch.cuda.is_available(): if torch.cuda.is_available():
device = 'cuda' device = 'cuda'
else: self.log('Using CUDA (Compute Unified Device Architecture)')
self.log('GPU not available. CUDA is not installed?')
if 'cuda' in device: if 'cuda' in device:
assert torch.cuda.is_available() assert torch.cuda.is_available()