mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-03-15 14:09:36 +00:00
More friendly error messages for corrupted safetensors files.
This commit is contained in:
parent
1210d094c7
commit
13fd4d6e45
@ -50,7 +50,16 @@ def load_torch_file(ckpt, safe_load=False, device=None):
|
|||||||
if device is None:
|
if device is None:
|
||||||
device = torch.device("cpu")
|
device = torch.device("cpu")
|
||||||
if ckpt.lower().endswith(".safetensors") or ckpt.lower().endswith(".sft"):
|
if ckpt.lower().endswith(".safetensors") or ckpt.lower().endswith(".sft"):
|
||||||
sd = safetensors.torch.load_file(ckpt, device=device.type)
|
try:
|
||||||
|
sd = safetensors.torch.load_file(ckpt, device=device.type)
|
||||||
|
except Exception as e:
|
||||||
|
if len(e.args) > 0:
|
||||||
|
message = e.args[0]
|
||||||
|
if "HeaderTooLarge" in message:
|
||||||
|
raise ValueError("{}\n\nFile path: {}\n\nThe safetensors file is corrupt or invalid. Make sure this is actually a safetensors file and not a ckpt or pt or other filetype.".format(message, ckpt))
|
||||||
|
if "MetadataIncompleteBuffer" in message:
|
||||||
|
raise ValueError("{}\n\nFile path: {}\n\nThe safetensors file is incomplete. Check the file size and make sure you have copied/downloaded it correctly.".format(message, ckpt))
|
||||||
|
raise e
|
||||||
else:
|
else:
|
||||||
if safe_load or ALWAYS_SAFE_LOAD:
|
if safe_load or ALWAYS_SAFE_LOAD:
|
||||||
pl_sd = torch.load(ckpt, map_location=device, weights_only=True)
|
pl_sd = torch.load(ckpt, map_location=device, weights_only=True)
|
||||||
|
Loading…
Reference in New Issue
Block a user