fix: GPU memory check threshold (#387)

2noise · Jun 21, 2024 · c42852e · c42852e
1 parent 56227f4
commit c42852e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ChatTTS/core.py b/ChatTTS/core.py
@@ -95,7 +95,7 @@ def _load(
         compile: bool = True,
     ):
         if not device:
-            device = select_device(4096)
+            device = select_device(4095)
             self.logger.log(logging.INFO, f'use {device}')
 
         if vocos_config_path: