fix(dl): from hf

2noise · Aug 28, 2024 · 024f93e · 024f93e
1 parent 4e4bbc6
commit 024f93e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ChatTTS/core.py b/ChatTTS/core.py
@@ -97,7 +97,7 @@ def download_models(
                 try:
                     download_path = snapshot_download(
                         repo_id="2Noise/ChatTTS",
-                        allow_patterns=["*.pt", "*.yaml", "*.json"],
+                        allow_patterns=["*.pt", "*.yaml", "*.json", "*.safetensors"],
                     )
                 except:
                     download_path = None