Spaces:
Sleeping
Sleeping
Commit
·
3e2e994
1
Parent(s):
d092760
initialize pipeline outside function
Browse files- kokoro_tts.py +7 -6
kokoro_tts.py
CHANGED
@@ -19,9 +19,15 @@ except:
|
|
19 |
return pipeline(text, voice=voice, speed=speed, split_pattern=r"\.")
|
20 |
|
21 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
22 |
def generate_audio(
|
23 |
text,
|
24 |
-
pipeline=KPipeline(lang_code="a"),
|
25 |
voice="af_heart",
|
26 |
speed=1,
|
27 |
save_segments=False,
|
@@ -40,11 +46,6 @@ def generate_audio(
|
|
40 |
Returns:
|
41 |
numpy.ndarray: Combined audio data at 24kHz sample rate
|
42 |
"""
|
43 |
-
try:
|
44 |
-
pipeline = pipeline.to("cuda")
|
45 |
-
except:
|
46 |
-
logging.warning("CUDA not available, using CPU")
|
47 |
-
|
48 |
generator = get_generator(pipeline, text, voice, speed, split_pattern="\.")
|
49 |
|
50 |
all_audio = []
|
|
|
19 |
return pipeline(text, voice=voice, speed=speed, split_pattern=r"\.")
|
20 |
|
21 |
|
22 |
+
pipeline = KPipeline(lang_code="a")
|
23 |
+
try:
|
24 |
+
pipeline = pipeline.to("cuda")
|
25 |
+
except:
|
26 |
+
logging.warning("CUDA not available, using CPU")
|
27 |
+
|
28 |
+
|
29 |
def generate_audio(
|
30 |
text,
|
|
|
31 |
voice="af_heart",
|
32 |
speed=1,
|
33 |
save_segments=False,
|
|
|
46 |
Returns:
|
47 |
numpy.ndarray: Combined audio data at 24kHz sample rate
|
48 |
"""
|
|
|
|
|
|
|
|
|
|
|
49 |
generator = get_generator(pipeline, text, voice, speed, split_pattern="\.")
|
50 |
|
51 |
all_audio = []
|