diff --git a/GPT_SoVITS/AR/models/t2s_lightning_module.py b/GPT_SoVITS/AR/models/t2s_lightning_module.py index 0696c35..97f3a08 100644 --- a/GPT_SoVITS/AR/models/t2s_lightning_module.py +++ b/GPT_SoVITS/AR/models/t2s_lightning_module.py @@ -28,7 +28,7 @@ class Text2SemanticLightningModule(LightningModule): self.load_state_dict( torch.load( pretrained_s1, - map_location="cpu", + map_location="cpu", weights_only=False, )["weight"], ) )