Fix .sft file loading (they are safetensors files).

NeoWorldTeam · Aug 1, 2024 · eb96c3b · eb96c3b
1 parent 5f98de7
commit eb96c3b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/comfy/utils.py b/comfy/utils.py
@@ -11,7 +11,7 @@
 def load_torch_file(ckpt, safe_load=False, device=None):
     if device is None:
         device = torch.device("cpu")
-    if ckpt.lower().endswith(".safetensors"):
+    if ckpt.lower().endswith(".safetensors") or ckpt.lower().endswith(".sft"):
         sd = safetensors.torch.load_file(ckpt, device=device.type)
     else:
         if safe_load: