Support old pytorch.

2 years ago · 00a9189e30
parent 191af3ef71
commit 00a9189e30
1 changed files with 4 additions and 1 deletions
--- a/comfy/sd1_clip.py
+++ b/comfy/sd1_clip.py
@ -186,7 +186,10 @@ def load_embed(embedding_name, embedding_directory):
        import safetensors.torch
        embed = safetensors.torch.load_file(embed_path, device="cpu")
    else:
-        embed = torch.load(embed_path, weights_only=True, map_location="cpu")
+        if 'weights_only' in torch.load.__code__.co_varnames:
+            embed = torch.load(embed_path, weights_only=True, map_location="cpu")
+        else:
+            embed = torch.load(embed_path, map_location="cpu")
    if 'string_to_param' in embed:
        values = embed['string_to_param'].values()
    else: