From aca7dc8db698e5885fe9dd4850bacfa757b429b1 Mon Sep 17 00:00:00 2001
From: Gabriel de Marmiesse <gabrieldemarmiesse@gmail.com>
Date: Thu, 12 Feb 2026 16:11:13 +0100
Subject: [PATCH] Remove cudagraph reference (#123)

---
 pocket_tts/main.py | 8 --------
 1 file changed, 8 deletions(-)

diff --git a/pocket_tts/main.py b/pocket_tts/main.py
index 0c03591..6b4de4f 100644
--- a/pocket_tts/main.py
+++ b/pocket_tts/main.py
@@ -238,10 +238,6 @@ def generate(
     ] = MAX_TOKEN_PER_CHUNK,
 ):
     """Generate speech using Kyutai Pocket TTS."""
-    if "cuda" in device:
-        # Cuda graphs capturing does not play nice with multithreading.
-        os.environ["NO_CUDA_GRAPH"] = "1"
-
     log_level = logging.ERROR if quiet else logging.INFO
     with enable_logging("pocket_tts", log_level):
         if text == "-":
@@ -342,10 +338,6 @@ def export_voice(
         logger.info(f"✅ Successfully exported voice '{voice}' to '{out_path}'")
         return True
 
-    if "cuda" in device:
-        # Cuda graphs capturing does not play nice with multithreading.
-        os.environ["NO_CUDA_GRAPH"] = "1"
-
     log_level = logging.ERROR if quiet else logging.INFO
     success_count = 0