wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-09 11:16:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: xtts not taking into account device flag (#2951)

Browse Source 
* fix: xtts not taking into account device flag

* Style changes

---------

Co-authored-by: Julian Weber <julian.weber@hotmail.fr>
This commit is contained in:
loupzeur
2023-09-20 09:57:02 +02:00
committed by GitHub
co-authored by Julian Weber
parent 335ae63e01
commit da8b6bbce1
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
TTS/tts/models/xtts.py
+1 -1
View File
@@ -642,7 +642,7 @@ class Xtts(BaseTTS):
self.init_models()
if eval:
self.gpt.init_gpt_for_inference(kv_cache=self.args.kv_cache)
self.load_state_dict(load_fsspec(model_path)["model"], strict=strict)
self.load_state_dict(load_fsspec(model_path, map_location=self.device)["model"], strict=strict)
if eval:
self.gpt.init_gpt_for_inference(kv_cache=self.args.kv_cache)