valenti/spark-pocket-tts
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Remove cudagraph reference (#123)

Browse Source
This commit is contained in:
Gabriel de Marmiesse
2026-02-12 16:11:13 +01:00
committed by GitHub
parent 5d8b15c75e
commit aca7dc8db6
1 changed files with 0 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pocket_tts/main.py
-8
View File
@@ -238,10 +238,6 @@ def generate(
] = MAX_TOKEN_PER_CHUNK, ] = MAX_TOKEN_PER_CHUNK,
): ):
"""Generate speech using Kyutai Pocket TTS.""" """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 log_level = logging.ERROR if quiet else logging.INFO
with enable_logging("pocket_tts", log_level): with enable_logging("pocket_tts", log_level):
if text == "-": if text == "-":
@@ -342,10 +338,6 @@ def export_voice(
logger.info(f"✅ Successfully exported voice '{voice}' to '{out_path}'") logger.info(f"✅ Successfully exported voice '{voice}' to '{out_path}'")
return True 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 log_level = logging.ERROR if quiet else logging.INFO
success_count = 0 success_count = 0