From d71781fb15ea03984e22717d81448926affb8570 Mon Sep 17 00:00:00 2001
From: 34j <55338215+34j@users.noreply.github.com>
Date: Sat, 6 Apr 2024 16:36:17 +0900
Subject: [PATCH] fix(utils): save model to cpu

---
 src/so_vits_svc_fork/utils.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/so_vits_svc_fork/utils.py b/src/so_vits_svc_fork/utils.py
index 8f36a333..c2814c48 100644
--- a/src/so_vits_svc_fork/utils.py
+++ b/src/so_vits_svc_fork/utils.py
@@ -315,6 +315,7 @@ def save_checkpoint(
     iteration: int,
     checkpoint_path: Path | str,
 ) -> None:
+    model = model.to("cpu")
     LOG.info(
         "Saving model and optimizer state at epoch {} to {}".format(
             iteration, checkpoint_path