From b843804a75d2b08ee72b3cc6a5844ec986feed30 Mon Sep 17 00:00:00 2001
From: tsukumi <tsukumijima@users.noreply.github.com>
Date: Thu, 19 Dec 2024 13:51:52 +0900
Subject: [PATCH] Fix: TTSModel.unload() did not work in PyTorch-independent
 environments

---
 style_bert_vits2/tts_model.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/style_bert_vits2/tts_model.py b/style_bert_vits2/tts_model.py
index 1351db7..ab0dd4c 100644
--- a/style_bert_vits2/tts_model.py
+++ b/style_bert_vits2/tts_model.py
@@ -230,12 +230,12 @@ class TTSModel:
         PyTorch モデルの場合は CUDA メモリも解放される。
         """
 
-        import torch
-
         start_time = time.time()
 
         # PyTorch 推論時
         if self.net_g is not None:
+            import torch
+
             del self.net_g
             self.net_g = None