unknown commited on
Commit
0de2e53
·
1 Parent(s): 7e758f4

fix extend

Browse files
src/f5_tts/train/finetune_gradio.py CHANGED
@@ -801,10 +801,12 @@ def vocab_extend(project_name, symbols, model_type):
801
  return "Symbols are okay no need to extend."
802
 
803
  size_vocab = len(vocab)
804
-
805
  for item in miss_symbols:
806
  vocab.append(item)
807
 
 
 
808
  with open(file_vocab_project, "w", encoding="utf-8") as f:
809
  f.write("\n".join(vocab))
810
 
 
801
  return "Symbols are okay no need to extend."
802
 
803
  size_vocab = len(vocab)
804
+ vocab.pop()
805
  for item in miss_symbols:
806
  vocab.append(item)
807
 
808
+ vocab.append("")
809
+
810
  with open(file_vocab_project, "w", encoding="utf-8") as f:
811
  f.write("\n".join(vocab))
812