fix(utils): save model to cpu

voicepaw · Apr 6, 2024 · d71781f · d71781f
1 parent 3f537b0
commit d71781f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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