Remove emojis #28 and bump to 0.3.2 (#29)

This commit is contained in:
hexgrad
2025-01-31 10:14:14 -08:00
committed by GitHub
parent e74290bf5a
commit 396766a5b0
3 changed files with 9 additions and 9 deletions

View File

@@ -1,4 +1,4 @@
__version__ = '0.3.1' __version__ = '0.3.2'
from .model import KModel from .model import KModel
from .pipeline import KPipeline from .pipeline import KPipeline

View File

@@ -86,7 +86,7 @@ class KPipeline:
try: try:
fallback = espeak.EspeakFallback(british=lang_code=='b') fallback = espeak.EspeakFallback(british=lang_code=='b')
except Exception as e: except Exception as e:
print('⚠️ WARNING: EspeakFallback not enabled. OOD words will be skipped.', e) print('WARNING: EspeakFallback not enabled. OOD words will be skipped.', e)
fallback = None fallback = None
self.g2p = en.G2P(trf=trf, british=lang_code=='b', fallback=fallback) self.g2p = en.G2P(trf=trf, british=lang_code=='b', fallback=fallback)
elif lang_code == 'j': elif lang_code == 'j':
@@ -94,18 +94,18 @@ class KPipeline:
from misaki import ja from misaki import ja
self.g2p = ja.JAG2P() self.g2p = ja.JAG2P()
except ImportError: except ImportError:
print("ERROR: You need to `pip install misaki[ja]` to use lang_code='j'") print("ERROR: You need to `pip install misaki[ja]` to use lang_code='j'")
raise raise
elif lang_code == 'z': elif lang_code == 'z':
try: try:
from misaki import zh from misaki import zh
self.g2p = zh.ZHG2P() self.g2p = zh.ZHG2P()
except ImportError: except ImportError:
print("ERROR: You need to `pip install misaki[zh]` to use lang_code='z'") print("ERROR: You need to `pip install misaki[zh]` to use lang_code='z'")
raise raise
else: else:
language = LANG_CODES[lang_code] language = LANG_CODES[lang_code]
print(f"⚠️ WARNING: Using EspeakG2P(language='{language}'). Chunking logic not yet implemented, so long texts may be truncated unless you split them with '\\n'.") print(f"WARNING: Using EspeakG2P(language='{language}'). Chunking logic not yet implemented, so long texts may be truncated unless you split them with '\\n'.")
self.g2p = espeak.EspeakG2P(language=language) self.g2p = espeak.EspeakG2P(language=language)
def load_single_voice(self, voice: str): def load_single_voice(self, voice: str):
@@ -118,7 +118,7 @@ class KPipeline:
if not voice.startswith(self.lang_code): if not voice.startswith(self.lang_code):
v = LANG_CODES.get(voice, voice) v = LANG_CODES.get(voice, voice)
p = LANG_CODES.get(self.lang_code, self.lang_code) p = LANG_CODES.get(self.lang_code, self.lang_code)
print(f'⚠️ WARNING: Language mismatch, loading {v} voice into {p} pipeline.') print(f'WARNING: Language mismatch, loading {v} voice into {p} pipeline.')
pack = torch.load(f, weights_only=True) pack = torch.load(f, weights_only=True)
self.voices[voice] = pack self.voices[voice] = pack
return pack return pack
@@ -217,7 +217,7 @@ class KPipeline:
if not ps: if not ps:
continue continue
elif len(ps) > 510: elif len(ps) > 510:
print(f"⚠️ WARNING: Unexpected len(ps) == {len(ps)} > 510 and ps == '{ps}'") print(f"WARNING: Unexpected len(ps) == {len(ps)} > 510 and ps == '{ps}'")
ps = ps[:510] ps = ps[:510]
yield gs, ps, KPipeline.infer(model, ps, pack, speed) yield gs, ps, KPipeline.infer(model, ps, pack, speed)
else: else:
@@ -225,6 +225,6 @@ class KPipeline:
if not ps: if not ps:
continue continue
elif len(ps) > 510: elif len(ps) > 510:
print(f'⚠️ WARNING: Truncating len(ps) == {len(ps)} > 510') print(f'WARNING: Truncating len(ps) == {len(ps)} > 510')
ps = ps[:510] ps = ps[:510]
yield graphemes, ps, KPipeline.infer(model, ps, pack, speed) yield graphemes, ps, KPipeline.infer(model, ps, pack, speed)

View File

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
setup( setup(
name='kokoro', name='kokoro',
version='0.3.1', version='0.3.2',
packages=find_packages(), packages=find_packages(),
install_requires=[ install_requires=[
'huggingface_hub', 'huggingface_hub',