Fix: torch_data_type

Artrajz · Nov 27, 2024 · e42797e · e42797e
1 parent 658c386
commit e42797e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/manager/model_handler.py b/manager/model_handler.py
@@ -125,7 +125,8 @@ def __init__(self, device=config.system.device):
         self.device = device
         self.ssl_model = None
 
-        if config.bert_vits2_config.torch_data_type.lower() in ["float16", "fp16"]:
+        torch_data_type = config.bert_vits2_config.torch_data_type
+        if torch_data_type and torch_data_type.lower() in ["float16", "fp16"]:
             self.torch_dtype = torch.float16
         else:
             self.torch_dtype = None