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

* fix: xtts not taking into account device flag * Style changes --------- Co-authored-by: Julian Weber <[email protected]>
coqui-ai · Sep 20, 2023 · da8b6bb · da8b6bb
1 parent 335ae63
commit da8b6bb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/TTS/tts/models/xtts.py b/TTS/tts/models/xtts.py
@@ -642,7 +642,7 @@ def load_checkpoint(
         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)