From e1278fa925cf59350bae76dc3d0c59a0e9564789 Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Thu, 25 May 2023 13:30:59 -0400
Subject: [PATCH] Support old pytorch versions that don't have weights_only.

---
 comfy/utils.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/comfy/utils.py b/comfy/utils.py
index 300eda6a..d58320b4 100644
--- a/comfy/utils.py
+++ b/comfy/utils.py
@@ -6,6 +6,10 @@ def load_torch_file(ckpt, safe_load=False):
         import safetensors.torch
         sd = safetensors.torch.load_file(ckpt, device="cpu")
     else:
+        if safe_load:
+            if not 'weights_only' in torch.load.__code__.co_varnames:
+                print("Warning torch.load doesn't support weights_only on this pytorch version, loading unsafely.")
+                safe_load = False
         if safe_load:
             pl_sd = torch.load(ckpt, map_location="cpu", weights_only=True)
         else: