chore: unify file names in utils (#402)

2noise · Jun 22, 2024 · 12cc9f6 · 12cc9f6
1 parent 3e0d87b
commit 12cc9f6
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 3 deletions.
diff --git a/ChatTTS/core.py b/ChatTTS/core.py
@@ -12,8 +12,8 @@
 
 from .model.dvae import DVAE
 from .model.gpt import GPT_warpper
-from .utils.gpu_utils import select_device
-from .utils.infer_utils import count_invalid_characters, detect_language, apply_character_map, apply_half2full_map, HomophonesReplacer
+from .utils.gpu import select_device
+from .utils.infer import count_invalid_characters, detect_language, apply_character_map, apply_half2full_map, HomophonesReplacer
 from .utils.io import get_latest_modified_file, del_all
 from .infer.api import refine_text, infer_code
 from .utils.download import check_all_assets, download_all_assets

diff --git a/ChatTTS/infer/api.py b/ChatTTS/infer/api.py
@@ -3,7 +3,7 @@
 import torch.nn.functional as F
 from transformers.generation import TopKLogitsWarper, TopPLogitsWarper
 
-from ..utils.infer_utils import CustomRepetitionPenaltyLogitsProcessorRepeat
+from ..utils.infer import CustomRepetitionPenaltyLogitsProcessorRepeat
 from ..utils.io import del_all
 from ..model.gpt import GPT_warpper
 

diff --git a/ChatTTS/utils/gpu_utils.py → ChatTTS/utils/gpu.py b/ChatTTS/utils/gpu_utils.py → ChatTTS/utils/gpu.py
diff --git a/ChatTTS/utils/infer_utils.py → ChatTTS/utils/infer.py b/ChatTTS/utils/infer_utils.py → ChatTTS/utils/infer.py