From daa36a5f7bf84af885c72350ea66e1aa941b70d7 Mon Sep 17 00:00:00 2001 From: Ethanfel Date: Thu, 9 Apr 2026 01:47:47 +0200 Subject: [PATCH] fix(bigvgan-trainer): clone target tensor to exit inference mode before backward Clips loaded outside torch.inference_mode(False) are inference tensors. Autograd cannot save them for backward. .clone() creates a normal tensor, same fix pattern as selva_lora_trainer's dist.mode().clone(). Co-Authored-By: Claude Sonnet 4.6 --- nodes/selva_bigvgan_trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nodes/selva_bigvgan_trainer.py b/nodes/selva_bigvgan_trainer.py index 26cae86..b86d8fc 100644 --- a/nodes/selva_bigvgan_trainer.py +++ b/nodes/selva_bigvgan_trainer.py @@ -232,7 +232,7 @@ class SelvaBigvganTrainer: start = random.randint(0, clip.shape[0] - segment_samples) batch.append(clip[start : start + segment_samples]) - target_flat = torch.stack(batch).to(device, dtype) # [B, T] + target_flat = torch.stack(batch).to(device, dtype).clone() # [B, T] target_wav = target_flat.unsqueeze(1) # [B, 1, T] # Fixed target mel (no grad needed here)