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