Browse Source

Safetensors isn't optional anymore.

pull/701/head
comfyanonymous 2 years ago
parent
commit
679bd2845a
  1. 11
      folder_paths.py

11
folder_paths.py

@ -1,14 +1,7 @@
import os import os
supported_ckpt_extensions = set(['.ckpt', '.pth']) supported_ckpt_extensions = set(['.ckpt', '.pth', '.safetensors'])
supported_pt_extensions = set(['.ckpt', '.pt', '.bin', '.pth']) supported_pt_extensions = set(['.ckpt', '.pt', '.bin', '.pth', '.safetensors'])
try:
import safetensors.torch
supported_ckpt_extensions.add('.safetensors')
supported_pt_extensions.add('.safetensors')
except:
print("Could not import safetensors, safetensors support disabled.")
folder_names_and_paths = {} folder_names_and_paths = {}

Loading…
Cancel
Save