Remove cudagraph reference (#123)

This commit is contained in:
Gabriel de Marmiesse
2026-02-12 16:11:13 +01:00
committed by GitHub
parent 5d8b15c75e
commit aca7dc8db6
-8
View File
@@ -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