From cba58fff0bfebfc81fbe678bb80491890a3df14a Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Wed, 15 Jan 2025 04:32:23 -0500
Subject: [PATCH] Remove unsafe embedding load for very old pytorch.

---
 comfy/sd1_clip.py | 11 ++++-------
 1 file changed, 4 insertions(+), 7 deletions(-)

diff --git a/comfy/sd1_clip.py b/comfy/sd1_clip.py
index 95d41c30f..85518afd9 100644
--- a/comfy/sd1_clip.py
+++ b/comfy/sd1_clip.py
@@ -388,13 +388,10 @@ def load_embed(embedding_name, embedding_directory, embedding_size, embed_key=No
             import safetensors.torch
             embed = safetensors.torch.load_file(embed_path, device="cpu")
         else:
-            if 'weights_only' in torch.load.__code__.co_varnames:
-                try:
-                    embed = torch.load(embed_path, weights_only=True, map_location="cpu")
-                except:
-                    embed_out = safe_load_embed_zip(embed_path)
-            else:
-                embed = torch.load(embed_path, map_location="cpu")
+            try:
+                embed = torch.load(embed_path, weights_only=True, map_location="cpu")
+            except:
+                embed_out = safe_load_embed_zip(embed_path)
     except Exception:
         logging.warning("{}\n\nerror loading embedding, skipping loading: {}".format(traceback.format_exc(), embedding_name))
         return None