Style changes

coqui-ai · Sep 19, 2023 · 95d3fc8 · 95d3fc8
1 parent 8468acf
commit 95d3fc8
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,self.device)["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)