update Gradient Checkpointing to reduce VRAM usage (#2040)

* update Gradient Checkpointing to reduce VRAM usage * fix inference
2025-02-12 23:00:34 +08:00
parent 86acb7a89d
commit c2b3298bed
5 changed files with 33 additions and 16 deletions
--- a/GPT_SoVITS/configs/s2.json
+++ b/GPT_SoVITS/configs/s2.json
@@ -18,7 +18,8 @@
    "warmup_epochs": 0,
    "c_mel": 45,
    "c_kl": 1.0,
-    "text_low_lr_rate": 0.4
+    "text_low_lr_rate": 0.4, 
+    "grad_ckpt": false
  },
  "data": {
    "max_wav_value": 32768.0,