search

jianchang512 / ChatTTS-ui

一个简单的本地网页界面，使用ChatTTS将文字合成为语音，同时支持对外提供API接口。A simple native web interface that uses ChatTTS to synthesize text into speech, along with support for external API interfaces.
https://pyvideotrans.com
Other
5.1k stars 556 forks source link

修复使用huggingface下载模型时CHATTTS_DIR没被定义的问题 #67

Closed fuyuwei01 closed 1 month ago

fuyuwei01 commented 1 month ago

从 huggingface.co下载模型时，CHATTTS_DIR没有被定义。并且环境变量好像也没切成功（可能是我的环境问题），模型还是下载到了默认的文件夹中。我参照ChatTTS.core.Chat.load_models的逻辑重写了这段代码。

jianchang512 commented 1 month ago

os.environ['HF_HUB_CACHE']=MODEL_DIR

os.environ['HF_ASSETS_CACHE']=MODEL_DIR

这两个不要删，默认缓存到当前目录下的midels里，而不是家目录里。

对于小白用户那个目录难找，而且有的用户使用下载工具下载后解压的，家目录不方便

以及存当前目录下，打包时可以直接包含模型，方便做整合包

fuyuwei01 commented 1 month ago

os.environ['HF_HUB_CACHE']=MODEL_DIR #os.environ['HF_ASSETS_CACHE']=MODEL_DIR

这两个不要删，默认缓存到当前目录下的midels里，而不是家目录里。

对于小白用户那个目录难找，而且有的用户使用下载工具下载后解压的，家目录不方便

以及存当前目录下，打包时可以直接包含模型，方便做整合包

了解了，或许可以直接用huggingface_hub.snapshot_download直接下到MODEL_DIR中。

jianchang512 commented 1 month ago

模型保存到当前的models下，对小白用户来说比较友好

改下重新提交吧