From ec1d6d66cf78898b7446cfef15732d535fc9aa6a Mon Sep 17 00:00:00 2001
From: Eren G??lge <egolge@coqui.ai>
Date: Wed, 24 May 2023 00:17:25 +0200
Subject: [PATCH] Pass cpu

---
 TTS/tts/utils/fairseq.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/TTS/tts/utils/fairseq.py b/TTS/tts/utils/fairseq.py
index 99c081ae..b1012958 100644
--- a/TTS/tts/utils/fairseq.py
+++ b/TTS/tts/utils/fairseq.py
@@ -2,7 +2,7 @@ import torch
 
 
 def rehash_fairseq_vits_checkpoint(checkpoint_file):
-    chk = torch.load(checkpoint_file)["model"]
+    chk = torch.load(checkpoint_file, map_location=torch.device('cpu'))["model"]
     new_chk = {}
     for k, v in chk.items():
         if "enc_p." in k: