fix(dl): from hf

fumiama · fumiama · commit 024f93e36e7d · 2024-08-28T22:52:40.000+08:00
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

Original file line number	Diff line number	Diff line change
`@@ -97,7 +97,7 @@ def download_models(`
`97`	`97`	`try:`
`98`	`98`	`download_path = snapshot_download(`
`99`	`99`	`repo_id="2Noise/ChatTTS",`
`100`		`- allow_patterns=[".pt", ".yaml", "*.json"],`
	`100`	`+ allow_patterns=[".pt", ".yaml", ".json", ".safetensors"],`
`101`	`101`	`)`
`102`	`102`	`except:`
`103`	`103`	`download_path = None`