xinference 1.9.1__py3-none-any.whl → 1.10.1__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of xinference might be problematic. Click here for more details.

Files changed (334) hide show
  1. xinference/_version.py +3 -3
  2. xinference/api/restful_api.py +400 -3
  3. xinference/client/restful/async_restful_client.py +20 -3
  4. xinference/client/restful/restful_client.py +20 -3
  5. xinference/constants.py +2 -0
  6. xinference/core/supervisor.py +111 -49
  7. xinference/core/worker.py +10 -0
  8. xinference/deploy/cmdline.py +15 -0
  9. xinference/model/audio/core.py +26 -6
  10. xinference/model/audio/indextts2.py +166 -0
  11. xinference/model/audio/kokoro.py +1 -1
  12. xinference/model/audio/kokoro_zh.py +124 -0
  13. xinference/model/audio/model_spec.json +58 -1
  14. xinference/model/embedding/sentence_transformers/core.py +4 -4
  15. xinference/model/embedding/vllm/core.py +7 -1
  16. xinference/model/image/model_spec.json +71 -3
  17. xinference/model/image/stable_diffusion/core.py +13 -4
  18. xinference/model/llm/__init__.py +4 -0
  19. xinference/model/llm/core.py +10 -0
  20. xinference/model/llm/llama_cpp/core.py +1 -0
  21. xinference/model/llm/llm_family.json +503 -21
  22. xinference/model/llm/llm_family.py +1 -0
  23. xinference/model/llm/mlx/core.py +52 -33
  24. xinference/model/llm/sglang/core.py +32 -55
  25. xinference/model/llm/tool_parsers/__init__.py +58 -0
  26. xinference/model/llm/tool_parsers/abstract_tool_parser.py +33 -0
  27. xinference/model/llm/tool_parsers/deepseek_r1_tool_parser.py +190 -0
  28. xinference/model/llm/tool_parsers/deepseek_v3_tool_parser.py +145 -0
  29. xinference/model/llm/tool_parsers/glm4_tool_parser.py +123 -0
  30. xinference/model/llm/tool_parsers/llama3_tool_parser.py +77 -0
  31. xinference/model/llm/tool_parsers/qwen_tool_parser.py +320 -0
  32. xinference/model/llm/transformers/core.py +1 -1
  33. xinference/model/llm/transformers/multimodal/qwen2_vl.py +34 -8
  34. xinference/model/llm/utils.py +138 -53
  35. xinference/model/llm/vllm/core.py +95 -78
  36. xinference/thirdparty/audiotools/__init__.py +10 -0
  37. xinference/thirdparty/audiotools/core/__init__.py +4 -0
  38. xinference/thirdparty/audiotools/core/audio_signal.py +1682 -0
  39. xinference/thirdparty/audiotools/core/display.py +194 -0
  40. xinference/thirdparty/audiotools/core/dsp.py +390 -0
  41. xinference/thirdparty/audiotools/core/effects.py +647 -0
  42. xinference/thirdparty/audiotools/core/ffmpeg.py +211 -0
  43. xinference/thirdparty/audiotools/core/loudness.py +320 -0
  44. xinference/thirdparty/audiotools/core/playback.py +252 -0
  45. xinference/thirdparty/audiotools/core/templates/__init__.py +0 -0
  46. xinference/thirdparty/audiotools/core/templates/headers.html +322 -0
  47. xinference/thirdparty/audiotools/core/templates/pandoc.css +407 -0
  48. xinference/thirdparty/audiotools/core/templates/widget.html +52 -0
  49. xinference/thirdparty/audiotools/core/util.py +671 -0
  50. xinference/thirdparty/audiotools/core/whisper.py +97 -0
  51. xinference/thirdparty/audiotools/data/__init__.py +3 -0
  52. xinference/thirdparty/audiotools/data/datasets.py +517 -0
  53. xinference/thirdparty/audiotools/data/preprocess.py +81 -0
  54. xinference/thirdparty/audiotools/data/transforms.py +1592 -0
  55. xinference/thirdparty/audiotools/metrics/__init__.py +6 -0
  56. xinference/thirdparty/audiotools/metrics/distance.py +131 -0
  57. xinference/thirdparty/audiotools/metrics/quality.py +159 -0
  58. xinference/thirdparty/audiotools/metrics/spectral.py +247 -0
  59. xinference/thirdparty/audiotools/ml/__init__.py +5 -0
  60. xinference/thirdparty/audiotools/ml/accelerator.py +184 -0
  61. xinference/thirdparty/audiotools/ml/decorators.py +440 -0
  62. xinference/thirdparty/audiotools/ml/experiment.py +90 -0
  63. xinference/thirdparty/audiotools/ml/layers/__init__.py +2 -0
  64. xinference/thirdparty/audiotools/ml/layers/base.py +328 -0
  65. xinference/thirdparty/audiotools/ml/layers/spectral_gate.py +127 -0
  66. xinference/thirdparty/audiotools/post.py +140 -0
  67. xinference/thirdparty/audiotools/preference.py +600 -0
  68. xinference/thirdparty/indextts/BigVGAN/ECAPA_TDNN.py +656 -0
  69. xinference/thirdparty/indextts/BigVGAN/__init__.py +0 -0
  70. xinference/thirdparty/indextts/BigVGAN/activations.py +122 -0
  71. xinference/thirdparty/indextts/BigVGAN/alias_free_activation/__init__.py +0 -0
  72. xinference/thirdparty/indextts/BigVGAN/alias_free_activation/cuda/.gitignore +1 -0
  73. xinference/thirdparty/indextts/BigVGAN/alias_free_activation/cuda/__init__.py +0 -0
  74. xinference/thirdparty/indextts/BigVGAN/alias_free_activation/cuda/activation1d.py +76 -0
  75. xinference/thirdparty/indextts/BigVGAN/alias_free_activation/cuda/anti_alias_activation.cpp +23 -0
  76. xinference/thirdparty/indextts/BigVGAN/alias_free_activation/cuda/anti_alias_activation_cuda.cu +256 -0
  77. xinference/thirdparty/indextts/BigVGAN/alias_free_activation/cuda/compat.h +29 -0
  78. xinference/thirdparty/indextts/BigVGAN/alias_free_activation/cuda/load.py +121 -0
  79. xinference/thirdparty/indextts/BigVGAN/alias_free_activation/cuda/type_shim.h +92 -0
  80. xinference/thirdparty/indextts/BigVGAN/alias_free_activation/torch/__init__.py +6 -0
  81. xinference/thirdparty/indextts/BigVGAN/alias_free_activation/torch/act.py +31 -0
  82. xinference/thirdparty/indextts/BigVGAN/alias_free_activation/torch/filter.py +102 -0
  83. xinference/thirdparty/indextts/BigVGAN/alias_free_activation/torch/resample.py +58 -0
  84. xinference/thirdparty/indextts/BigVGAN/alias_free_torch/__init__.py +6 -0
  85. xinference/thirdparty/indextts/BigVGAN/alias_free_torch/act.py +29 -0
  86. xinference/thirdparty/indextts/BigVGAN/alias_free_torch/filter.py +96 -0
  87. xinference/thirdparty/indextts/BigVGAN/alias_free_torch/resample.py +49 -0
  88. xinference/thirdparty/indextts/BigVGAN/bigvgan.py +534 -0
  89. xinference/thirdparty/indextts/BigVGAN/models.py +451 -0
  90. xinference/thirdparty/indextts/BigVGAN/nnet/CNN.py +546 -0
  91. xinference/thirdparty/indextts/BigVGAN/nnet/__init__.py +0 -0
  92. xinference/thirdparty/indextts/BigVGAN/nnet/linear.py +89 -0
  93. xinference/thirdparty/indextts/BigVGAN/nnet/normalization.py +670 -0
  94. xinference/thirdparty/indextts/BigVGAN/utils.py +101 -0
  95. xinference/thirdparty/indextts/__init__.py +0 -0
  96. xinference/thirdparty/indextts/cli.py +65 -0
  97. xinference/thirdparty/indextts/gpt/__init__.py +0 -0
  98. xinference/thirdparty/indextts/gpt/conformer/__init__.py +0 -0
  99. xinference/thirdparty/indextts/gpt/conformer/attention.py +312 -0
  100. xinference/thirdparty/indextts/gpt/conformer/embedding.py +163 -0
  101. xinference/thirdparty/indextts/gpt/conformer/subsampling.py +348 -0
  102. xinference/thirdparty/indextts/gpt/conformer_encoder.py +520 -0
  103. xinference/thirdparty/indextts/gpt/model.py +713 -0
  104. xinference/thirdparty/indextts/gpt/model_v2.py +747 -0
  105. xinference/thirdparty/indextts/gpt/perceiver.py +317 -0
  106. xinference/thirdparty/indextts/gpt/transformers_beam_search.py +1013 -0
  107. xinference/thirdparty/indextts/gpt/transformers_generation_utils.py +4747 -0
  108. xinference/thirdparty/indextts/gpt/transformers_gpt2.py +1878 -0
  109. xinference/thirdparty/indextts/gpt/transformers_modeling_utils.py +5525 -0
  110. xinference/thirdparty/indextts/infer.py +690 -0
  111. xinference/thirdparty/indextts/infer_v2.py +739 -0
  112. xinference/thirdparty/indextts/s2mel/dac/__init__.py +16 -0
  113. xinference/thirdparty/indextts/s2mel/dac/__main__.py +36 -0
  114. xinference/thirdparty/indextts/s2mel/dac/model/__init__.py +4 -0
  115. xinference/thirdparty/indextts/s2mel/dac/model/base.py +294 -0
  116. xinference/thirdparty/indextts/s2mel/dac/model/dac.py +400 -0
  117. xinference/thirdparty/indextts/s2mel/dac/model/discriminator.py +228 -0
  118. xinference/thirdparty/indextts/s2mel/dac/model/encodec.py +320 -0
  119. xinference/thirdparty/indextts/s2mel/dac/nn/__init__.py +3 -0
  120. xinference/thirdparty/indextts/s2mel/dac/nn/layers.py +33 -0
  121. xinference/thirdparty/indextts/s2mel/dac/nn/loss.py +368 -0
  122. xinference/thirdparty/indextts/s2mel/dac/nn/quantize.py +339 -0
  123. xinference/thirdparty/indextts/s2mel/dac/utils/__init__.py +123 -0
  124. xinference/thirdparty/indextts/s2mel/dac/utils/decode.py +95 -0
  125. xinference/thirdparty/indextts/s2mel/dac/utils/encode.py +94 -0
  126. xinference/thirdparty/indextts/s2mel/hf_utils.py +12 -0
  127. xinference/thirdparty/indextts/s2mel/modules/alias_free_torch/__init__.py +5 -0
  128. xinference/thirdparty/indextts/s2mel/modules/alias_free_torch/act.py +29 -0
  129. xinference/thirdparty/indextts/s2mel/modules/alias_free_torch/filter.py +96 -0
  130. xinference/thirdparty/indextts/s2mel/modules/alias_free_torch/resample.py +57 -0
  131. xinference/thirdparty/indextts/s2mel/modules/audio.py +82 -0
  132. xinference/thirdparty/indextts/s2mel/modules/bigvgan/activations.py +120 -0
  133. xinference/thirdparty/indextts/s2mel/modules/bigvgan/alias_free_activation/cuda/__init__.py +0 -0
  134. xinference/thirdparty/indextts/s2mel/modules/bigvgan/alias_free_activation/cuda/activation1d.py +77 -0
  135. xinference/thirdparty/indextts/s2mel/modules/bigvgan/alias_free_activation/cuda/anti_alias_activation.cpp +23 -0
  136. xinference/thirdparty/indextts/s2mel/modules/bigvgan/alias_free_activation/cuda/anti_alias_activation_cuda.cu +246 -0
  137. xinference/thirdparty/indextts/s2mel/modules/bigvgan/alias_free_activation/cuda/compat.h +29 -0
  138. xinference/thirdparty/indextts/s2mel/modules/bigvgan/alias_free_activation/cuda/load.py +86 -0
  139. xinference/thirdparty/indextts/s2mel/modules/bigvgan/alias_free_activation/cuda/type_shim.h +92 -0
  140. xinference/thirdparty/indextts/s2mel/modules/bigvgan/alias_free_activation/torch/__init__.py +6 -0
  141. xinference/thirdparty/indextts/s2mel/modules/bigvgan/alias_free_activation/torch/act.py +30 -0
  142. xinference/thirdparty/indextts/s2mel/modules/bigvgan/alias_free_activation/torch/filter.py +101 -0
  143. xinference/thirdparty/indextts/s2mel/modules/bigvgan/alias_free_activation/torch/resample.py +58 -0
  144. xinference/thirdparty/indextts/s2mel/modules/bigvgan/bigvgan.py +492 -0
  145. xinference/thirdparty/indextts/s2mel/modules/bigvgan/config.json +63 -0
  146. xinference/thirdparty/indextts/s2mel/modules/bigvgan/env.py +18 -0
  147. xinference/thirdparty/indextts/s2mel/modules/bigvgan/meldataset.py +354 -0
  148. xinference/thirdparty/indextts/s2mel/modules/bigvgan/utils.py +99 -0
  149. xinference/thirdparty/indextts/s2mel/modules/campplus/DTDNN.py +115 -0
  150. xinference/thirdparty/indextts/s2mel/modules/campplus/classifier.py +70 -0
  151. xinference/thirdparty/indextts/s2mel/modules/campplus/layers.py +253 -0
  152. xinference/thirdparty/indextts/s2mel/modules/commons.py +632 -0
  153. xinference/thirdparty/indextts/s2mel/modules/diffusion_transformer.py +257 -0
  154. xinference/thirdparty/indextts/s2mel/modules/encodec.py +292 -0
  155. xinference/thirdparty/indextts/s2mel/modules/flow_matching.py +171 -0
  156. xinference/thirdparty/indextts/s2mel/modules/gpt_fast/generate.py +436 -0
  157. xinference/thirdparty/indextts/s2mel/modules/gpt_fast/model.py +360 -0
  158. xinference/thirdparty/indextts/s2mel/modules/gpt_fast/quantize.py +622 -0
  159. xinference/thirdparty/indextts/s2mel/modules/hifigan/f0_predictor.py +55 -0
  160. xinference/thirdparty/indextts/s2mel/modules/hifigan/generator.py +454 -0
  161. xinference/thirdparty/indextts/s2mel/modules/layers.py +354 -0
  162. xinference/thirdparty/indextts/s2mel/modules/length_regulator.py +141 -0
  163. xinference/thirdparty/indextts/s2mel/modules/openvoice/__init__.py +0 -0
  164. xinference/thirdparty/indextts/s2mel/modules/openvoice/api.py +186 -0
  165. xinference/thirdparty/indextts/s2mel/modules/openvoice/attentions.py +465 -0
  166. xinference/thirdparty/indextts/s2mel/modules/openvoice/checkpoints_v2/converter/config.json +57 -0
  167. xinference/thirdparty/indextts/s2mel/modules/openvoice/commons.py +160 -0
  168. xinference/thirdparty/indextts/s2mel/modules/openvoice/mel_processing.py +183 -0
  169. xinference/thirdparty/indextts/s2mel/modules/openvoice/models.py +499 -0
  170. xinference/thirdparty/indextts/s2mel/modules/openvoice/modules.py +598 -0
  171. xinference/thirdparty/indextts/s2mel/modules/openvoice/openvoice_app.py +275 -0
  172. xinference/thirdparty/indextts/s2mel/modules/openvoice/se_extractor.py +153 -0
  173. xinference/thirdparty/indextts/s2mel/modules/openvoice/transforms.py +209 -0
  174. xinference/thirdparty/indextts/s2mel/modules/openvoice/utils.py +194 -0
  175. xinference/thirdparty/indextts/s2mel/modules/quantize.py +229 -0
  176. xinference/thirdparty/indextts/s2mel/modules/rmvpe.py +631 -0
  177. xinference/thirdparty/indextts/s2mel/modules/vocos/__init__.py +4 -0
  178. xinference/thirdparty/indextts/s2mel/modules/vocos/heads.py +164 -0
  179. xinference/thirdparty/indextts/s2mel/modules/vocos/helpers.py +71 -0
  180. xinference/thirdparty/indextts/s2mel/modules/vocos/loss.py +114 -0
  181. xinference/thirdparty/indextts/s2mel/modules/vocos/models.py +118 -0
  182. xinference/thirdparty/indextts/s2mel/modules/vocos/modules.py +213 -0
  183. xinference/thirdparty/indextts/s2mel/modules/vocos/pretrained.py +51 -0
  184. xinference/thirdparty/indextts/s2mel/modules/vocos/spectral_ops.py +192 -0
  185. xinference/thirdparty/indextts/s2mel/modules/wavenet.py +174 -0
  186. xinference/thirdparty/indextts/s2mel/optimizers.py +96 -0
  187. xinference/thirdparty/indextts/s2mel/wav2vecbert_extract.py +148 -0
  188. xinference/thirdparty/indextts/utils/__init__.py +0 -0
  189. xinference/thirdparty/indextts/utils/arch_util.py +120 -0
  190. xinference/thirdparty/indextts/utils/checkpoint.py +34 -0
  191. xinference/thirdparty/indextts/utils/common.py +121 -0
  192. xinference/thirdparty/indextts/utils/feature_extractors.py +50 -0
  193. xinference/thirdparty/indextts/utils/front.py +536 -0
  194. xinference/thirdparty/indextts/utils/maskgct/models/codec/__init__.py +0 -0
  195. xinference/thirdparty/indextts/utils/maskgct/models/codec/amphion_codec/codec.py +427 -0
  196. xinference/thirdparty/indextts/utils/maskgct/models/codec/amphion_codec/quantize/__init__.py +11 -0
  197. xinference/thirdparty/indextts/utils/maskgct/models/codec/amphion_codec/quantize/factorized_vector_quantize.py +150 -0
  198. xinference/thirdparty/indextts/utils/maskgct/models/codec/amphion_codec/quantize/lookup_free_quantize.py +77 -0
  199. xinference/thirdparty/indextts/utils/maskgct/models/codec/amphion_codec/quantize/residual_vq.py +177 -0
  200. xinference/thirdparty/indextts/utils/maskgct/models/codec/amphion_codec/quantize/vector_quantize.py +401 -0
  201. xinference/thirdparty/indextts/utils/maskgct/models/codec/amphion_codec/vocos.py +881 -0
  202. xinference/thirdparty/indextts/utils/maskgct/models/codec/codec_dataset.py +264 -0
  203. xinference/thirdparty/indextts/utils/maskgct/models/codec/codec_inference.py +515 -0
  204. xinference/thirdparty/indextts/utils/maskgct/models/codec/codec_sampler.py +126 -0
  205. xinference/thirdparty/indextts/utils/maskgct/models/codec/codec_trainer.py +166 -0
  206. xinference/thirdparty/indextts/utils/maskgct/models/codec/facodec/__init__.py +0 -0
  207. xinference/thirdparty/indextts/utils/maskgct/models/codec/facodec/alias_free_torch/__init__.py +5 -0
  208. xinference/thirdparty/indextts/utils/maskgct/models/codec/facodec/alias_free_torch/act.py +29 -0
  209. xinference/thirdparty/indextts/utils/maskgct/models/codec/facodec/alias_free_torch/filter.py +96 -0
  210. xinference/thirdparty/indextts/utils/maskgct/models/codec/facodec/alias_free_torch/resample.py +57 -0
  211. xinference/thirdparty/indextts/utils/maskgct/models/codec/facodec/facodec_dataset.py +98 -0
  212. xinference/thirdparty/indextts/utils/maskgct/models/codec/facodec/facodec_inference.py +137 -0
  213. xinference/thirdparty/indextts/utils/maskgct/models/codec/facodec/facodec_trainer.py +776 -0
  214. xinference/thirdparty/indextts/utils/maskgct/models/codec/facodec/modules/JDC/__init__.py +1 -0
  215. xinference/thirdparty/indextts/utils/maskgct/models/codec/facodec/modules/JDC/bst.t7 +0 -0
  216. xinference/thirdparty/indextts/utils/maskgct/models/codec/facodec/modules/JDC/model.py +219 -0
  217. xinference/thirdparty/indextts/utils/maskgct/models/codec/facodec/modules/attentions.py +437 -0
  218. xinference/thirdparty/indextts/utils/maskgct/models/codec/facodec/modules/commons.py +331 -0
  219. xinference/thirdparty/indextts/utils/maskgct/models/codec/facodec/modules/gradient_reversal.py +35 -0
  220. xinference/thirdparty/indextts/utils/maskgct/models/codec/facodec/modules/layers.py +460 -0
  221. xinference/thirdparty/indextts/utils/maskgct/models/codec/facodec/modules/quantize.py +741 -0
  222. xinference/thirdparty/indextts/utils/maskgct/models/codec/facodec/modules/style_encoder.py +110 -0
  223. xinference/thirdparty/indextts/utils/maskgct/models/codec/facodec/modules/wavenet.py +224 -0
  224. xinference/thirdparty/indextts/utils/maskgct/models/codec/facodec/optimizer.py +104 -0
  225. xinference/thirdparty/indextts/utils/maskgct/models/codec/kmeans/repcodec_model.py +210 -0
  226. xinference/thirdparty/indextts/utils/maskgct/models/codec/kmeans/vocos.py +850 -0
  227. xinference/thirdparty/indextts/utils/maskgct/models/codec/melvqgan/melspec.py +108 -0
  228. xinference/thirdparty/indextts/utils/maskgct/models/codec/ns3_codec/README.md +216 -0
  229. xinference/thirdparty/indextts/utils/maskgct/models/codec/ns3_codec/__init__.py +6 -0
  230. xinference/thirdparty/indextts/utils/maskgct/models/codec/ns3_codec/alias_free_torch/__init__.py +5 -0
  231. xinference/thirdparty/indextts/utils/maskgct/models/codec/ns3_codec/alias_free_torch/act.py +29 -0
  232. xinference/thirdparty/indextts/utils/maskgct/models/codec/ns3_codec/alias_free_torch/filter.py +96 -0
  233. xinference/thirdparty/indextts/utils/maskgct/models/codec/ns3_codec/alias_free_torch/resample.py +57 -0
  234. xinference/thirdparty/indextts/utils/maskgct/models/codec/ns3_codec/facodec.py +1222 -0
  235. xinference/thirdparty/indextts/utils/maskgct/models/codec/ns3_codec/gradient_reversal.py +35 -0
  236. xinference/thirdparty/indextts/utils/maskgct/models/codec/ns3_codec/melspec.py +102 -0
  237. xinference/thirdparty/indextts/utils/maskgct/models/codec/ns3_codec/quantize/__init__.py +7 -0
  238. xinference/thirdparty/indextts/utils/maskgct/models/codec/ns3_codec/quantize/fvq.py +116 -0
  239. xinference/thirdparty/indextts/utils/maskgct/models/codec/ns3_codec/quantize/rvq.py +87 -0
  240. xinference/thirdparty/indextts/utils/maskgct/models/codec/ns3_codec/transformer.py +234 -0
  241. xinference/thirdparty/indextts/utils/maskgct/models/codec/speechtokenizer/model.py +184 -0
  242. xinference/thirdparty/indextts/utils/maskgct/models/codec/speechtokenizer/modules/__init__.py +27 -0
  243. xinference/thirdparty/indextts/utils/maskgct/models/codec/speechtokenizer/modules/conv.py +346 -0
  244. xinference/thirdparty/indextts/utils/maskgct/models/codec/speechtokenizer/modules/lstm.py +46 -0
  245. xinference/thirdparty/indextts/utils/maskgct/models/codec/speechtokenizer/modules/norm.py +37 -0
  246. xinference/thirdparty/indextts/utils/maskgct/models/codec/speechtokenizer/modules/quantization/__init__.py +14 -0
  247. xinference/thirdparty/indextts/utils/maskgct/models/codec/speechtokenizer/modules/quantization/ac.py +317 -0
  248. xinference/thirdparty/indextts/utils/maskgct/models/codec/speechtokenizer/modules/quantization/core_vq.py +388 -0
  249. xinference/thirdparty/indextts/utils/maskgct/models/codec/speechtokenizer/modules/quantization/distrib.py +135 -0
  250. xinference/thirdparty/indextts/utils/maskgct/models/codec/speechtokenizer/modules/quantization/vq.py +125 -0
  251. xinference/thirdparty/indextts/utils/maskgct/models/codec/speechtokenizer/modules/seanet.py +414 -0
  252. xinference/thirdparty/indextts/utils/maskgct/models/codec/vevo/vevo_repcodec.py +592 -0
  253. xinference/thirdparty/indextts/utils/maskgct/models/tts/maskgct/ckpt/wav2vec2bert_stats.pt +0 -0
  254. xinference/thirdparty/indextts/utils/maskgct/models/tts/maskgct/llama_nar.py +650 -0
  255. xinference/thirdparty/indextts/utils/maskgct/models/tts/maskgct/maskgct_s2a.py +503 -0
  256. xinference/thirdparty/indextts/utils/maskgct_utils.py +259 -0
  257. xinference/thirdparty/indextts/utils/text_utils.py +41 -0
  258. xinference/thirdparty/indextts/utils/typical_sampling.py +30 -0
  259. xinference/thirdparty/indextts/utils/utils.py +93 -0
  260. xinference/thirdparty/indextts/utils/webui_utils.py +42 -0
  261. xinference/thirdparty/indextts/utils/xtransformers.py +1247 -0
  262. xinference/thirdparty/indextts/vqvae/__init__.py +0 -0
  263. xinference/thirdparty/indextts/vqvae/xtts_dvae.py +395 -0
  264. xinference/types.py +105 -2
  265. xinference/ui/gradio/media_interface.py +66 -8
  266. xinference/ui/web/ui/build/asset-manifest.json +6 -6
  267. xinference/ui/web/ui/build/index.html +1 -1
  268. xinference/ui/web/ui/build/static/css/main.5ea97072.css +2 -0
  269. xinference/ui/web/ui/build/static/css/main.5ea97072.css.map +1 -0
  270. xinference/ui/web/ui/build/static/js/main.d192c4f3.js +3 -0
  271. xinference/ui/web/ui/build/static/js/{main.1086c759.js.LICENSE.txt → main.d192c4f3.js.LICENSE.txt} +0 -7
  272. xinference/ui/web/ui/build/static/js/main.d192c4f3.js.map +1 -0
  273. xinference/ui/web/ui/node_modules/.cache/babel-loader/089c38df5f52348d212ed868dda5c518a42e0c2762caed4175487c0405830c35.json +1 -0
  274. xinference/ui/web/ui/node_modules/.cache/babel-loader/2b6e3a5b6eb2c5c5f2d007e68cd46c372721cd52bf63508adcdb21ecf79241d8.json +1 -0
  275. xinference/ui/web/ui/node_modules/.cache/babel-loader/2d887825fd07a56f872eda4420da25fba0b5b62a23bdcc6c6da1a5281887f618.json +1 -0
  276. xinference/ui/web/ui/node_modules/.cache/babel-loader/4001f9c3e64e73a4f2158826650c174a59d5e3f89ddecddf17cbb6bb688cc4ca.json +1 -0
  277. xinference/ui/web/ui/node_modules/.cache/babel-loader/4a7018a69e6b7f90fc313248c2aa86f2a8f1eb1db120df586047a8023549b44b.json +1 -0
  278. xinference/ui/web/ui/node_modules/.cache/babel-loader/64b12aaa1c1d1bf53820ada8a63769067c0ccc5aab46b32348eb1917ae7f2a11.json +1 -0
  279. xinference/ui/web/ui/node_modules/.cache/babel-loader/7275b67c78ec76ce38a686bb8a576d8c9cecf54e1573614c84859d538efb9be5.json +1 -0
  280. xinference/ui/web/ui/node_modules/.cache/babel-loader/a68b6ee3b31eadc051fb95ce8f8ccb9c2e8b52c60f290dbab545a1917e065282.json +1 -0
  281. xinference/ui/web/ui/node_modules/.cache/babel-loader/ae8771cc37693feb160fa8727231312a0c54ef2d1d1ca893be568cd70016ca7e.json +1 -0
  282. xinference/ui/web/ui/node_modules/.cache/babel-loader/bb4e8722d2d41d87f1fce3661bc8937bffe9448e231fc5f0462630849e851592.json +1 -0
  283. xinference/ui/web/ui/node_modules/.cache/babel-loader/be6aada1ee4adc2bbf65dbe56d17db32bb3b5478be05d6b527805a8ba6cfb2b9.json +1 -0
  284. xinference/ui/web/ui/node_modules/.cache/babel-loader/de91c352653c233cf0cb6674e6e04049a44fd0e1156560de65d5c4620521391e.json +1 -0
  285. xinference/ui/web/ui/node_modules/.cache/babel-loader/e85f7002fc325c83b9c9cd8a1619e5b3ebc701d30e811afc284b88e6ae710cb5.json +1 -0
  286. xinference/ui/web/ui/node_modules/.cache/babel-loader/e8b603c78944bf3d213639078bfe155ff5c0dfa4048a93cbb967cad6a4eb4ff3.json +1 -0
  287. xinference/ui/web/ui/node_modules/.cache/babel-loader/f05535160a508b2a312de546a6de234776c613db276479ea4253c0b1bdeeb7d6.json +1 -0
  288. xinference/ui/web/ui/node_modules/.cache/babel-loader/f09ba9e11106bd59a0de10cc85c55084097729dcab575f43dfcf07375961ed87.json +1 -0
  289. xinference/ui/web/ui/node_modules/.cache/babel-loader/f995a2425dfb0822fd07127f66ffe9b026883bc156b402eb8bd0b83d52460a93.json +1 -0
  290. xinference/ui/web/ui/node_modules/.package-lock.json +0 -33
  291. xinference/ui/web/ui/package-lock.json +0 -34
  292. xinference/ui/web/ui/package.json +0 -1
  293. xinference/ui/web/ui/src/locales/en.json +9 -3
  294. xinference/ui/web/ui/src/locales/ja.json +9 -3
  295. xinference/ui/web/ui/src/locales/ko.json +9 -3
  296. xinference/ui/web/ui/src/locales/zh.json +9 -3
  297. {xinference-1.9.1.dist-info → xinference-1.10.1.dist-info}/METADATA +24 -4
  298. {xinference-1.9.1.dist-info → xinference-1.10.1.dist-info}/RECORD +302 -76
  299. xinference/ui/web/ui/build/static/css/main.013f296b.css +0 -2
  300. xinference/ui/web/ui/build/static/css/main.013f296b.css.map +0 -1
  301. xinference/ui/web/ui/build/static/js/main.1086c759.js +0 -3
  302. xinference/ui/web/ui/build/static/js/main.1086c759.js.map +0 -1
  303. xinference/ui/web/ui/node_modules/.cache/babel-loader/0b0f77000cc1b482ca091cfbcae511dfe02f08916971645fad21d0b1234d04a2.json +0 -1
  304. xinference/ui/web/ui/node_modules/.cache/babel-loader/1c5f8ff423a7c9202bea60b15680f04b1e9964b445b0da3f86c6ff70cf24e797.json +0 -1
  305. xinference/ui/web/ui/node_modules/.cache/babel-loader/44ce7993e344980e3ed4f13e8f69237d4a5dfc60e37ca6b54f51f8ee1357bd67.json +0 -1
  306. xinference/ui/web/ui/node_modules/.cache/babel-loader/4aec1cc414ac3ebb3481d3d915e4db597d9127de813291346eacb8554ab170d4.json +0 -1
  307. xinference/ui/web/ui/node_modules/.cache/babel-loader/644cfec52f3c57a6e222ce60f112237a1efefe9835efd9aad857a685f53d8eed.json +0 -1
  308. xinference/ui/web/ui/node_modules/.cache/babel-loader/663436f72af53fe0d72394f56d003fa4e0bba489e5bb4e483fd34b00f84637f7.json +0 -1
  309. xinference/ui/web/ui/node_modules/.cache/babel-loader/69db82ca9bfe27fe417cc6cf2b1716b09be9c6f0cd198530f12bfc60e801bbcf.json +0 -1
  310. xinference/ui/web/ui/node_modules/.cache/babel-loader/85087e27618d740c236bf159f30e0219db443ab55f0997388eed5fde6f9e90cc.json +0 -1
  311. xinference/ui/web/ui/node_modules/.cache/babel-loader/88b07838348864aa86c672be3bbca1e9f58f6f3a2881b32070ec27f4e7b449d1.json +0 -1
  312. xinference/ui/web/ui/node_modules/.cache/babel-loader/8b8cd408ccfbe115acef27ccfa5b233da8597131a2a5712add13e1e4d5d4504b.json +0 -1
  313. xinference/ui/web/ui/node_modules/.cache/babel-loader/a23824fe746b9c6ca5eee9159b5764d1ff1653c1d856288c0f75c742bbb0023b.json +0 -1
  314. xinference/ui/web/ui/node_modules/.cache/babel-loader/a3eb18af328280b139693c9092dff2a0ef8c9a967e6c8956ceee0996611f1984.json +0 -1
  315. xinference/ui/web/ui/node_modules/.cache/babel-loader/bc1aacc65a102db325ca61bcd2f681e1ae22c36a1f1d98a6ff5e4ad49dc7544f.json +0 -1
  316. xinference/ui/web/ui/node_modules/.cache/babel-loader/c682fd521747c19dae437d83ce3235a306ce6b68e24a117bc57c27ebb8d1f1ca.json +0 -1
  317. xinference/ui/web/ui/node_modules/.cache/babel-loader/d5c224be7081f18cba1678b7874a9782eba895df004874ff8f243f94ba79942a.json +0 -1
  318. xinference/ui/web/ui/node_modules/.cache/babel-loader/f7f18bfb539b036a6a342176dd98a85df5057a884a8da978d679f2a0264883d0.json +0 -1
  319. xinference/ui/web/ui/node_modules/clipboard/.babelrc.json +0 -11
  320. xinference/ui/web/ui/node_modules/clipboard/.eslintrc.json +0 -24
  321. xinference/ui/web/ui/node_modules/clipboard/.prettierrc.json +0 -9
  322. xinference/ui/web/ui/node_modules/clipboard/bower.json +0 -18
  323. xinference/ui/web/ui/node_modules/clipboard/composer.json +0 -25
  324. xinference/ui/web/ui/node_modules/clipboard/package.json +0 -63
  325. xinference/ui/web/ui/node_modules/delegate/package.json +0 -31
  326. xinference/ui/web/ui/node_modules/good-listener/bower.json +0 -11
  327. xinference/ui/web/ui/node_modules/good-listener/package.json +0 -35
  328. xinference/ui/web/ui/node_modules/select/bower.json +0 -13
  329. xinference/ui/web/ui/node_modules/select/package.json +0 -29
  330. xinference/ui/web/ui/node_modules/tiny-emitter/package.json +0 -53
  331. {xinference-1.9.1.dist-info → xinference-1.10.1.dist-info}/WHEEL +0 -0
  332. {xinference-1.9.1.dist-info → xinference-1.10.1.dist-info}/entry_points.txt +0 -0
  333. {xinference-1.9.1.dist-info → xinference-1.10.1.dist-info}/licenses/LICENSE +0 -0
  334. {xinference-1.9.1.dist-info → xinference-1.10.1.dist-info}/top_level.txt +0 -0
@@ -0,0 +1,536 @@
1
+ # -*- coding: utf-8 -*-
2
+ import os
3
+ import traceback
4
+ import re
5
+ from typing import List, Union, overload
6
+ import warnings
7
+ from indextts.utils.common import tokenize_by_CJK_char, de_tokenized_by_CJK_char
8
+ from sentencepiece import SentencePieceProcessor
9
+
10
+
11
+ class TextNormalizer:
12
+ def __init__(self):
13
+ self.zh_normalizer = None
14
+ self.en_normalizer = None
15
+ self.char_rep_map = {
16
+ ":": ",",
17
+ ";": ",",
18
+ ";": ",",
19
+ ",": ",",
20
+ "。": ".",
21
+ "!": "!",
22
+ "?": "?",
23
+ "\n": " ",
24
+ "·": "-",
25
+ "、": ",",
26
+ "...": "…",
27
+ ",,,": "…",
28
+ ",,,": "…",
29
+ "……": "…",
30
+ "“": "'",
31
+ "”": "'",
32
+ '"': "'",
33
+ "‘": "'",
34
+ "’": "'",
35
+ "(": "'",
36
+ ")": "'",
37
+ "(": "'",
38
+ ")": "'",
39
+ "《": "'",
40
+ "》": "'",
41
+ "【": "'",
42
+ "】": "'",
43
+ "[": "'",
44
+ "]": "'",
45
+ "—": "-",
46
+ "~": "-",
47
+ "~": "-",
48
+ "「": "'",
49
+ "」": "'",
50
+ ":": ",",
51
+ }
52
+ self.zh_char_rep_map = {
53
+ "$": ".",
54
+ **self.char_rep_map,
55
+ }
56
+
57
+ def match_email(self, email):
58
+ # 正则表达式匹配邮箱格式:数字英文@数字英文.英文
59
+ pattern = r"^[a-zA-Z0-9]+@[a-zA-Z0-9]+\.[a-zA-Z]+$"
60
+ return re.match(pattern, email) is not None
61
+
62
+ PINYIN_TONE_PATTERN = r"(?<![a-z])((?:[bpmfdtnlgkhjqxzcsryw]|[zcs]h)?(?:[aeiouüv]|[ae]i|u[aio]|ao|ou|i[aue]|[uüv]e|[uvü]ang?|uai|[aeiuv]n|[aeio]ng|ia[no]|i[ao]ng)|ng|er)([1-5])"
63
+ """
64
+ 匹配拼音声调格式:pinyin+数字,声调1-5,5表示轻声
65
+ 例如:xuan4, jve2, ying1, zhong4, shang5
66
+ 不匹配:beta1, voice2
67
+ """
68
+ NAME_PATTERN = r"[\u4e00-\u9fff]+(?:[-·—][\u4e00-\u9fff]+){1,2}"
69
+ """
70
+ 匹配人名,格式:中文·中文,中文·中文-中文
71
+ 例如:克里斯托弗·诺兰,约瑟夫·高登-莱维特
72
+ """
73
+
74
+ # 匹配常见英语缩写 's,仅用于替换为 is,不匹配所有 's
75
+ ENGLISH_CONTRACTION_PATTERN = r"(what|where|who|which|how|t?here|it|s?he|that|this)'s"
76
+
77
+
78
+ def use_chinese(self, s):
79
+ has_chinese = bool(re.search(r"[\u4e00-\u9fff]", s))
80
+ has_alpha = bool(re.search(r"[a-zA-Z]", s))
81
+ is_email = self.match_email(s)
82
+ if has_chinese or not has_alpha or is_email:
83
+ return True
84
+
85
+ has_pinyin = bool(re.search(TextNormalizer.PINYIN_TONE_PATTERN, s, re.IGNORECASE))
86
+ return has_pinyin
87
+
88
+ def load(self):
89
+ # print(os.path.join(os.path.dirname(os.path.abspath(__file__)), ".."))
90
+ # sys.path.append(model_dir)
91
+ import platform
92
+ if self.zh_normalizer is not None and self.en_normalizer is not None:
93
+ return
94
+ if platform.system() != "Linux": # Mac and Windows
95
+ from wetext import Normalizer
96
+
97
+ self.zh_normalizer = Normalizer(remove_erhua=False, lang="zh", operator="tn")
98
+ self.en_normalizer = Normalizer(lang="en", operator="tn")
99
+ else:
100
+ from tn.chinese.normalizer import Normalizer as NormalizerZh
101
+ from tn.english.normalizer import Normalizer as NormalizerEn
102
+ # use new cache dir for build tagger rules with disable remove_interjections and remove_erhua
103
+ cache_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "tagger_cache")
104
+ if not os.path.exists(cache_dir):
105
+ os.makedirs(cache_dir)
106
+ with open(os.path.join(cache_dir, ".gitignore"), "w") as f:
107
+ f.write("*\n")
108
+ self.zh_normalizer = NormalizerZh(
109
+ cache_dir=cache_dir, remove_interjections=False, remove_erhua=False, overwrite_cache=False
110
+ )
111
+ self.en_normalizer = NormalizerEn(overwrite_cache=False)
112
+
113
+ def normalize(self, text: str) -> str:
114
+ if not self.zh_normalizer or not self.en_normalizer:
115
+ print("Error, text normalizer is not initialized !!!")
116
+ return ""
117
+ if self.use_chinese(text):
118
+ text = re.sub(TextNormalizer.ENGLISH_CONTRACTION_PATTERN, r"\1 is", text, flags=re.IGNORECASE)
119
+ replaced_text, pinyin_list = self.save_pinyin_tones(text.rstrip())
120
+
121
+ replaced_text, original_name_list = self.save_names(replaced_text)
122
+ try:
123
+ result = self.zh_normalizer.normalize(replaced_text)
124
+ except Exception:
125
+ result = ""
126
+ print(traceback.format_exc())
127
+ # 恢复人名
128
+ result = self.restore_names(result, original_name_list)
129
+ # 恢复拼音声调
130
+ result = self.restore_pinyin_tones(result, pinyin_list)
131
+ pattern = re.compile("|".join(re.escape(p) for p in self.zh_char_rep_map.keys()))
132
+ result = pattern.sub(lambda x: self.zh_char_rep_map[x.group()], result)
133
+ else:
134
+ try:
135
+ text = re.sub(TextNormalizer.ENGLISH_CONTRACTION_PATTERN, r"\1 is", text, flags=re.IGNORECASE)
136
+ result = self.en_normalizer.normalize(text)
137
+ except Exception:
138
+ result = text
139
+ print(traceback.format_exc())
140
+ pattern = re.compile("|".join(re.escape(p) for p in self.char_rep_map.keys()))
141
+ result = pattern.sub(lambda x: self.char_rep_map[x.group()], result)
142
+ return result
143
+
144
+ def correct_pinyin(self, pinyin: str):
145
+ """
146
+ 将 jqx 的韵母为 u/ü 的拼音转换为 v
147
+ 如:ju -> jv , que -> qve, xün -> xvn
148
+ """
149
+ if pinyin[0] not in "jqxJQX":
150
+ return pinyin
151
+ # 匹配 jqx 的韵母为 u/ü 的拼音
152
+ pattern = r"([jqx])[uü](n|e|an)*(\d)"
153
+ repl = r"\g<1>v\g<2>\g<3>"
154
+ pinyin = re.sub(pattern, repl, pinyin, flags=re.IGNORECASE)
155
+ return pinyin.upper()
156
+
157
+ def save_names(self, original_text):
158
+ """
159
+ 替换人名为占位符 <n_a>、 <n_b>, ...
160
+ 例如:克里斯托弗·诺兰 -> <n_a>
161
+ """
162
+ # 人名
163
+ name_pattern = re.compile(TextNormalizer.NAME_PATTERN, re.IGNORECASE)
164
+ original_name_list = re.findall(name_pattern, original_text)
165
+ if len(original_name_list) == 0:
166
+ return (original_text, None)
167
+ original_name_list = list(set("".join(n) for n in original_name_list))
168
+ transformed_text = original_text
169
+ # 替换占位符 <n_a>、 <n_b>, ...
170
+ for i, name in enumerate(original_name_list):
171
+ number = chr(ord("a") + i)
172
+ transformed_text = transformed_text.replace(name, f"<n_{number}>")
173
+
174
+ return transformed_text, original_name_list
175
+
176
+ def restore_names(self, normalized_text, original_name_list):
177
+ """
178
+ 恢复人名为原来的文字
179
+ 例如:<n_a> -> original_name_list[0]
180
+ """
181
+ if not original_name_list or len(original_name_list) == 0:
182
+ return normalized_text
183
+
184
+ transformed_text = normalized_text
185
+ # 替换为占位符 <n_a>、 <n_b>, ...
186
+ for i, name in enumerate(original_name_list):
187
+ number = chr(ord("a") + i)
188
+ transformed_text = transformed_text.replace(f"<n_{number}>", name)
189
+ return transformed_text
190
+
191
+ def save_pinyin_tones(self, original_text):
192
+ """
193
+ 替换拼音声调为占位符 <pinyin_a>, <pinyin_b>, ...
194
+ 例如:xuan4 -> <pinyin_a>
195
+ """
196
+ # 声母韵母+声调数字
197
+ origin_pinyin_pattern = re.compile(TextNormalizer.PINYIN_TONE_PATTERN, re.IGNORECASE)
198
+ original_pinyin_list = re.findall(origin_pinyin_pattern, original_text)
199
+ if len(original_pinyin_list) == 0:
200
+ return (original_text, None)
201
+ original_pinyin_list = list(set("".join(p) for p in original_pinyin_list))
202
+ transformed_text = original_text
203
+ # 替换为占位符 <pinyin_a>, <pinyin_b>, ...
204
+ for i, pinyin in enumerate(original_pinyin_list):
205
+ number = chr(ord("a") + i)
206
+ transformed_text = transformed_text.replace(pinyin, f"<pinyin_{number}>")
207
+
208
+ # print("original_text: ", original_text)
209
+ # print("transformed_text: ", transformed_text)
210
+ return transformed_text, original_pinyin_list
211
+
212
+ def restore_pinyin_tones(self, normalized_text, original_pinyin_list):
213
+ """
214
+ 恢复拼音中的音调数字(1-5)为原来的拼音
215
+ 例如:<pinyin_a> -> original_pinyin_list[0]
216
+ """
217
+ if not original_pinyin_list or len(original_pinyin_list) == 0:
218
+ return normalized_text
219
+
220
+ transformed_text = normalized_text
221
+ # 替换占位符 <pinyin_a>, <pinyin_b>, ...
222
+ for i, pinyin in enumerate(original_pinyin_list):
223
+ number = chr(ord("a") + i)
224
+ pinyin = self.correct_pinyin(pinyin)
225
+ transformed_text = transformed_text.replace(f"<pinyin_{number}>", pinyin)
226
+ # print("normalized_text: ", normalized_text)
227
+ # print("transformed_text: ", transformed_text)
228
+ return transformed_text
229
+
230
+
231
+ class TextTokenizer:
232
+ def __init__(self, vocab_file: str, normalizer: TextNormalizer = None):
233
+ self.vocab_file = vocab_file
234
+ self.normalizer = normalizer
235
+
236
+ if self.vocab_file is None:
237
+ raise ValueError("vocab_file is None")
238
+ if not os.path.exists(self.vocab_file):
239
+ raise ValueError(f"vocab_file {self.vocab_file} does not exist")
240
+ if self.normalizer:
241
+ self.normalizer.load()
242
+ # 加载词表
243
+ self.sp_model = SentencePieceProcessor(model_file=self.vocab_file)
244
+
245
+ self.pre_tokenizers = [
246
+ # 预处理器
247
+ tokenize_by_CJK_char,
248
+ ]
249
+
250
+ @property
251
+ def vocab_size(self):
252
+ return self.sp_model.GetPieceSize()
253
+
254
+ @property
255
+ def unk_token(self):
256
+ return "<unk>"
257
+
258
+ @property
259
+ def pad_token(self):
260
+ return None
261
+
262
+ @property
263
+ def bos_token(self):
264
+ return "<s>"
265
+
266
+ @property
267
+ def eos_token(self):
268
+ return "</s>"
269
+
270
+ @property
271
+ def pad_token_id(self):
272
+ return -1
273
+
274
+ @property
275
+ def bos_token_id(self):
276
+ return 0
277
+
278
+ @property
279
+ def eos_token_id(self):
280
+ return 1
281
+
282
+ @property
283
+ def unk_token_id(self):
284
+ return self.sp_model.unk_id()
285
+
286
+ @property
287
+ def special_tokens_map(self):
288
+ return {
289
+ "unk_token": self.unk_token,
290
+ "pad_token": self.pad_token,
291
+ "bos_token": self.bos_token,
292
+ "eos_token": self.eos_token,
293
+ }
294
+
295
+ def get_vocab(self):
296
+ vocab = {self.convert_ids_to_tokens(i): i for i in range(self.vocab_size)}
297
+ return vocab
298
+
299
+ @overload
300
+ def convert_ids_to_tokens(self, ids: int) -> str: ...
301
+
302
+ @overload
303
+ def convert_ids_to_tokens(self, ids: List[int]) -> List[str]: ...
304
+
305
+ def convert_ids_to_tokens(self, ids: Union[List[int], int]):
306
+ return self.sp_model.IdToPiece(ids)
307
+
308
+ def convert_tokens_to_ids(self, tokens: Union[List[str], str]) -> List[int]:
309
+ if isinstance(tokens, str):
310
+ tokens = [tokens]
311
+ return [self.sp_model.PieceToId(token) for token in tokens]
312
+
313
+ def tokenize(self, text: str) -> List[str]:
314
+ return self.encode(text, out_type=str)
315
+
316
+ def encode(self, text: str, **kwargs):
317
+ if len(text) == 0:
318
+ return []
319
+ if len(text.strip()) == 1:
320
+ return self.sp_model.Encode(text, out_type=kwargs.pop("out_type", int), **kwargs)
321
+ # 预处理
322
+ if self.normalizer:
323
+ text = self.normalizer.normalize(text)
324
+ if len(self.pre_tokenizers) > 0:
325
+ for pre_tokenizer in self.pre_tokenizers:
326
+ text = pre_tokenizer(text)
327
+ return self.sp_model.Encode(text, out_type=kwargs.pop("out_type", int), **kwargs)
328
+
329
+ def batch_encode(self, texts: List[str], **kwargs):
330
+ # 预处理
331
+ if self.normalizer:
332
+ texts = [self.normalizer.normalize(text) for text in texts]
333
+ if len(self.pre_tokenizers) > 0:
334
+ for pre_tokenizer in self.pre_tokenizers:
335
+ texts = [pre_tokenizer(text) for text in texts]
336
+ return self.sp_model.Encode(texts, out_type=kwargs.pop("out_type", int), **kwargs)
337
+
338
+ def decode(self, ids: Union[List[int], int], do_lower_case=False, **kwargs):
339
+ if isinstance(ids, int):
340
+ ids = [ids]
341
+ decoded = self.sp_model.Decode(ids, out_type=kwargs.pop("out_type", str), **kwargs)
342
+ return de_tokenized_by_CJK_char(decoded, do_lower_case=do_lower_case)
343
+
344
+ @staticmethod
345
+ def split_segments_by_token(
346
+ tokenized_str: List[str], split_tokens: List[str], max_text_tokens_per_segment: int
347
+ ) -> List[List[str]]:
348
+ """
349
+ 将tokenize后的结果按特定token进一步分割
350
+ """
351
+ # 处理特殊情况
352
+ if len(tokenized_str) == 0:
353
+ return []
354
+ segments: List[List[str]] = []
355
+ current_segment = []
356
+ current_segment_tokens_len = 0
357
+ for i in range(len(tokenized_str)):
358
+ token = tokenized_str[i]
359
+ current_segment.append(token)
360
+ current_segment_tokens_len += 1
361
+ if current_segment_tokens_len <= max_text_tokens_per_segment:
362
+ if token in split_tokens and current_segment_tokens_len > 2:
363
+ if i < len(tokenized_str) - 1:
364
+ if tokenized_str[i + 1] in ["'", "▁'"]:
365
+ # 后续token是',则不切分
366
+ current_segment.append(tokenized_str[i + 1])
367
+ i += 1
368
+ segments.append(current_segment)
369
+ current_segment = []
370
+ current_segment_tokens_len = 0
371
+ continue
372
+ # 如果当前tokens的长度超过最大限制
373
+ if not ("," in split_tokens or "▁," in split_tokens ) and ("," in current_segment or "▁," in current_segment):
374
+ # 如果当前tokens中有,,则按,分割
375
+ sub_segments = TextTokenizer.split_segments_by_token(
376
+ current_segment, [",", "▁,"], max_text_tokens_per_segment=max_text_tokens_per_segment
377
+ )
378
+ elif "-" not in split_tokens and "-" in current_segment:
379
+ # 没有,,则按-分割
380
+ sub_segments = TextTokenizer.split_segments_by_token(
381
+ current_segment, ["-"], max_text_tokens_per_segment=max_text_tokens_per_segment
382
+ )
383
+ else:
384
+ # 按照长度分割
385
+ sub_segments = []
386
+ for j in range(0, len(current_segment), max_text_tokens_per_segment):
387
+ if j + max_text_tokens_per_segment < len(current_segment):
388
+ sub_segments.append(current_segment[j : j + max_text_tokens_per_segment])
389
+ else:
390
+ sub_segments.append(current_segment[j:])
391
+ warnings.warn(
392
+ f"The tokens length of segment exceeds limit: {max_text_tokens_per_segment}, "
393
+ f"Tokens in segment: {current_segment}."
394
+ "Maybe unexpected behavior",
395
+ RuntimeWarning,
396
+ )
397
+ segments.extend(sub_segments)
398
+ current_segment = []
399
+ current_segment_tokens_len = 0
400
+ if current_segment_tokens_len > 0:
401
+ assert current_segment_tokens_len <= max_text_tokens_per_segment
402
+ segments.append(current_segment)
403
+ # 如果相邻的句子加起来长度小于最大限制,则合并
404
+ merged_segments = []
405
+ for segment in segments:
406
+ if len(segment) == 0:
407
+ continue
408
+ if len(merged_segments) == 0:
409
+ merged_segments.append(segment)
410
+ elif len(merged_segments[-1]) + len(segment) <= max_text_tokens_per_segment:
411
+ merged_segments[-1] = merged_segments[-1] + segment
412
+ else:
413
+ merged_segments.append(segment)
414
+ return merged_segments
415
+
416
+ punctuation_marks_tokens = [
417
+ ".",
418
+ "!",
419
+ "?",
420
+ "▁.",
421
+ # "▁!", # unk
422
+ "▁?",
423
+ "▁...", # ellipsis
424
+ ]
425
+ def split_segments(self, tokenized: List[str], max_text_tokens_per_segment=120) -> List[List[str]]:
426
+ return TextTokenizer.split_segments_by_token(
427
+ tokenized, self.punctuation_marks_tokens, max_text_tokens_per_segment=max_text_tokens_per_segment
428
+ )
429
+
430
+
431
+ if __name__ == "__main__":
432
+ # 测试程序
433
+
434
+ text_normalizer = TextNormalizer()
435
+
436
+ cases = [
437
+ "IndexTTS 正式发布1.0版本了,效果666",
438
+ "晕XUAN4是一种GAN3觉",
439
+ "我爱你!",
440
+ "I love you!",
441
+ "“我爱你”的英语是“I love you”",
442
+ "2.5平方电线",
443
+ "共465篇,约315万字",
444
+ "2002年的第一场雪,下在了2003年",
445
+ "速度是10km/h",
446
+ "现在是北京时间2025年01月11日 20:00",
447
+ "他这条裤子是2012年买的,花了200块钱",
448
+ "电话:135-4567-8900",
449
+ "1键3连",
450
+ "他这条视频点赞3000+,评论1000+,收藏500+",
451
+ "这是1024元的手机,你要吗?",
452
+ "受不liao3你了",
453
+ "“衣裳”不读衣chang2,而是读衣shang5",
454
+ "最zhong4要的是:不要chong2蹈覆辙",
455
+ "不zuo1死就不会死",
456
+ "See you at 8:00 AM",
457
+ "8:00 AM 开会",
458
+ "Couting down 3, 2, 1, go!",
459
+ "数到3就开始:1、2、3",
460
+ "This sales for 2.5% off, only $12.5.",
461
+ "5G网络是4G网络的升级版,2G网络是3G网络的前身",
462
+ "苹果于2030/1/2发布新 iPhone 2X 系列手机,最低售价仅 ¥12999",
463
+ "这酒...里...有毒...",
464
+ # 异常case
465
+ "只有,,,才是最好的",
466
+ "babala2是什么?", # babala二是什么?
467
+ "用beta1测试", # 用beta一测试
468
+ "have you ever been to beta2?", # have you ever been to beta two?
469
+ "such as XTTS, CosyVoice2, Fish-Speech, and F5-TTS", # such as xtts,cosyvoice two,fish-speech,and f five-tts
470
+ "where's the money?", # where is the money?
471
+ "who's there?", # who is there?
472
+ "which's the best?", # which is the best?
473
+ "how's it going?", # how is it going?
474
+ "今天是个好日子 it's a good day", # 今天是个好日子 it is a good day
475
+ # 人名
476
+ "约瑟夫·高登-莱维特(Joseph Gordon-Levitt is an American actor)",
477
+ "蒂莫西·唐纳德·库克(英文名:Timothy Donald Cook),通称蒂姆·库克(Tim Cook),美国商业经理、工业工程师和工业开发商,现任苹果公司首席执行官。",
478
+ # 长句子
479
+ "《盗梦空间》是由美国华纳兄弟影片公司出品的电影,由克里斯托弗·诺兰执导并编剧,莱昂纳多·迪卡普里奥、玛丽昂·歌迪亚、约瑟夫·高登-莱维特、艾利奥特·佩吉、汤姆·哈迪等联袂主演,2010年7月16日在美国上映,2010年9月1日在中国内地上映,2020年8月28日在中国内地重映。影片剧情游走于梦境与现实之间,被定义为“发生在意识结构内的当代动作科幻片”,讲述了由莱昂纳多·迪卡普里奥扮演的造梦师,带领特工团队进入他人梦境,从他人的潜意识中盗取机密,并重塑他人梦境的故事。",
480
+ "清晨拉开窗帘,阳光洒在窗台的Bloomixy花艺礼盒上——薰衣草香薰蜡烛唤醒嗅觉,永生花束折射出晨露般光泽。设计师将“自然绽放美学”融入每个细节:手工陶瓷花瓶可作首饰收纳,香薰精油含依兰依兰舒缓配方。限量款附赠《365天插花灵感手册》,让每个平凡日子都有花开仪式感。\n宴会厅灯光暗下的刹那,Glimmeria星月系列耳坠开始发光——瑞士冷珐琅工艺让蓝宝石如银河流动,钛合金骨架仅3.2g无负重感。设计师秘密:内置微型重力感应器,随步伐产生0.01mm振幅,打造“行走的星光”。七夕限定礼盒含星座定制铭牌,让爱意如星辰永恒闪耀。",
481
+ "电影1:“黑暗骑士”(演员:克里斯蒂安·贝尔、希斯·莱杰;导演:克里斯托弗·诺兰);电影2:“盗梦空间”(演员:莱昂纳多·迪卡普里奥;导演:克里斯托弗·诺兰);电影3:“钢琴家”(演员:艾德里安·布洛迪;导演:罗曼·波兰斯基);电影4:“泰坦尼克号”(演员:莱昂纳多·迪卡普里奥;导演:詹姆斯·卡梅隆);电影5:“阿凡达”(演员:萨姆·沃辛顿;导演:詹姆斯·卡梅隆);电影6:“南方公园:大电影”(演员:马特·斯通、托马斯·艾恩格瑞;导演:特雷·帕克)",
482
+ ]
483
+ # 测试分词器
484
+ tokenizer = TextTokenizer(
485
+ vocab_file="checkpoints/bpe.model",
486
+ normalizer=text_normalizer,
487
+ )
488
+
489
+ codes = tokenizer.batch_encode(
490
+ cases,
491
+ out_type=int,
492
+ )
493
+
494
+ print(f"vocab_size: {tokenizer.vocab_size}")
495
+ # print(f"pad_token: {tokenizer.pad_token}, pad_token_id: {tokenizer.pad_token_id}")
496
+ print(f"bos_token: {tokenizer.bos_token}, bos_token_id: {tokenizer.bos_token_id}")
497
+ print(f"eos_token: {tokenizer.eos_token}, eos_token_id: {tokenizer.eos_token_id}")
498
+ print(f"unk_token: {tokenizer.unk_token}, unk_token_id: {tokenizer.unk_token_id}")
499
+ # 测试拼音 (8474-10201)
500
+ for id in range(8474, 10201):
501
+ pinyin = tokenizer.convert_ids_to_tokens(id)
502
+ if re.match(TextNormalizer.PINYIN_TONE_PATTERN, pinyin, re.IGNORECASE) is None:
503
+ print(f"{pinyin} should be matched")
504
+ for badcase in [
505
+ "beta1", "better1", "voice2", "bala2", "babala2", "hunger2"
506
+ ]:
507
+ if re.match(TextNormalizer.PINYIN_TONE_PATTERN, badcase, re.IGNORECASE) is not None:
508
+ print(f"{badcase} should not be matched!")
509
+ # 不应该有 unk_token_id
510
+ for t in set([*TextTokenizer.punctuation_marks_tokens, ",", "▁,", "-", "▁..."]):
511
+ tokens = tokenizer.convert_tokens_to_ids(t)
512
+ if tokenizer.unk_token_id in tokens:
513
+ print(f"Warning: {t} is unknown token")
514
+ print(f"`{t}`", "->", tokens, "->", tokenizer.convert_ids_to_tokens(tokens))
515
+ for ch in set(tokenizer.normalizer.zh_char_rep_map.values()):
516
+ # 测试 normalize后的字符能被分词器识别
517
+ print(f"`{ch}`", "->", tokenizer.sp_model.Encode(ch, out_type=str))
518
+ print(f"` {ch}`", "->", tokenizer.sp_model.Encode(f" {ch}", out_type=str))
519
+ max_text_tokens_per_segment=120
520
+ for i in range(len(cases)):
521
+ print(f"原始文本: {cases[i]}")
522
+ print(f"Normalized: {text_normalizer.normalize(cases[i])}")
523
+ tokens = tokenizer.tokenize(cases[i])
524
+ print("Tokenzied: ", ", ".join([f"`{t}`" for t in tokens]))
525
+ segments = tokenizer.split_segments(tokens, max_text_tokens_per_segment=max_text_tokens_per_segment)
526
+ print("Segments count:", len(segments))
527
+ if len(segments) > 1:
528
+ for j in range(len(segments)):
529
+ print(f" {j}, count:", len(segments[j]), ", tokens:", "".join(segments[j]))
530
+ if len(segments[j]) > max_text_tokens_per_segment:
531
+ print(f"Warning: segment {j} is too long, length: {len(segments[j])}")
532
+ #print(f"Token IDs (first 10): {codes[i][:10]}")
533
+ if tokenizer.unk_token in codes[i]:
534
+ print(f"Warning: `{cases[i]}` contains UNKNOWN token")
535
+ print(f"Decoded: {tokenizer.decode(codes[i], do_lower_case=True)}")
536
+ print("-" * 50)