From 36de06d6e31822e935224cbead694cdaefaa970d Mon Sep 17 00:00:00 2001
From: rwedge <roy.wedge@datacebo.com>
Date: Mon, 10 Feb 2025 10:56:05 -0600
Subject: [PATCH] torch load weights_only set to False

---
 ctgan/synthesizers/base.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ctgan/synthesizers/base.py b/ctgan/synthesizers/base.py
index add0dd7e..14b4f9c9 100644
--- a/ctgan/synthesizers/base.py
+++ b/ctgan/synthesizers/base.py
@@ -119,7 +119,7 @@ def save(self, path):
     def load(cls, path):
         """Load the model stored in the passed `path`."""
         device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
-        model = torch.load(path)
+        model = torch.load(path, weights_only=False)
         model.set_device(device)
         return model