Skip to content

Commit 36de06d

Browse files
committed
torch load weights_only set to False
1 parent dc2129f commit 36de06d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ctgan/synthesizers/base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -119,7 +119,7 @@ def save(self, path):
119119
def load(cls, path):
120120
"""Load the model stored in the passed `path`."""
121121
device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
122-
model = torch.load(path)
122+
model = torch.load(path, weights_only=False)
123123
model.set_device(device)
124124
return model
125125

0 commit comments

Comments
 (0)