Update utils.py

weights_only=True
neuroneural · Aug 12, 2024 · 940ee55 · 940ee55
1 parent 8f46a51
commit 940ee55
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mindfultensors/utils.py b/mindfultensors/utils.py
@@ -19,7 +19,7 @@ def qnormalize(img, qmin=0.01, qmax=0.99):
 
 def mtransform(tensor_binary):
     buffer = io.BytesIO(tensor_binary)
-    tensor = torch.load(buffer)
+    tensor = torch.load(buffer, weights_only=True)
     return tensor