cocktailpeanut commited on
Commit
5122d7b
Β·
1 Parent(s): f499b43
Files changed (1) hide show
  1. diffrhythm/g2p/g2p/__init__.py +3 -2
diffrhythm/g2p/g2p/__init__.py CHANGED
@@ -16,7 +16,7 @@ class PhonemeBpeTokenizer:
16
  def __init__(self, vacab_path="./diffrhythm/g2p/g2p/vocab.json"):
17
  self.lang2backend = {
18
  "zh": "cmn",
19
- "ja": "ja",
20
  "en": "en-us",
21
  "fr": "fr-fr",
22
  "ko": "ko",
@@ -29,7 +29,8 @@ class PhonemeBpeTokenizer:
29
  json_data = f.read()
30
  data = json.loads(json_data)
31
  self.vocab = data["vocab"]
32
- LangSegment.setfilters(["en", "zh", "ja", "ko", "fr", "de"])
 
33
 
34
  def int_text_tokenizers(self):
35
  for key, value in self.lang2backend.items():
 
16
  def __init__(self, vacab_path="./diffrhythm/g2p/g2p/vocab.json"):
17
  self.lang2backend = {
18
  "zh": "cmn",
19
+ # "ja": "ja",
20
  "en": "en-us",
21
  "fr": "fr-fr",
22
  "ko": "ko",
 
29
  json_data = f.read()
30
  data = json.loads(json_data)
31
  self.vocab = data["vocab"]
32
+ #LangSegment.setfilters(["en", "zh", "ja", "ko", "fr", "de"])
33
+ LangSegment.setfilters(["en", "zh", "ko", "fr", "de"])
34
 
35
  def int_text_tokenizers(self):
36
  for key, value in self.lang2backend.items():