From c42852ea384bfe1e22918a3bc769ee13fc1216ff Mon Sep 17 00:00:00 2001
From: ermian <nir311@163.com>
Date: Fri, 21 Jun 2024 15:18:43 +0800
Subject: [PATCH] fix: GPU memory check threshold (#387)

---
 ChatTTS/core.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ChatTTS/core.py b/ChatTTS/core.py
index f4cfada4c..3af5dd01d 100644
--- 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: