lambdaofgod commited on
Commit
3e2e994
·
1 Parent(s): d092760

initialize pipeline outside function

Browse files
Files changed (1) hide show
  1. 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 = []