From 5021a03de07dff3be6e89b57316984859aecbc5b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Eren=20G=C3=B6lge?= <egolge@coqui.ai>
Date: Wed, 11 May 2022 11:29:36 +0200
Subject: [PATCH] Use torch.no_grad for VITS inference

---
 TTS/tts/models/vits.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/TTS/tts/models/vits.py b/TTS/tts/models/vits.py
index 3bcd59a1..1c623f50 100644
--- a/TTS/tts/models/vits.py
+++ b/TTS/tts/models/vits.py
@@ -982,6 +982,7 @@ class Vits(BaseTTS):
             return aux_input["x_lengths"]
         return torch.tensor(x.shape[1:2]).to(x.device)
 
+    @torch.no_grad()
     def inference(
         self, x, aux_input={"x_lengths": None, "d_vectors": None, "speaker_ids": None, "language_ids": None}
     ):  # pylint: disable=dangerous-default-value