Safetensors isn't optional anymore.

2 years ago · 679bd2845a
parent eb4bd7711a
commit 679bd2845a
1 changed files with 2 additions and 9 deletions
--- a/folder_paths.py
+++ b/folder_paths.py
@ -1,14 +1,7 @@
 import os

-supported_ckpt_extensions = set(['.ckpt', '.pth'])
-supported_pt_extensions = set(['.ckpt', '.pt', '.bin', '.pth'])
-try:
-    import safetensors.torch
-    supported_ckpt_extensions.add('.safetensors')
-    supported_pt_extensions.add('.safetensors')
-except:
-    print("Could not import safetensors, safetensors support disabled.")
-
+supported_ckpt_extensions = set(['.ckpt', '.pth', '.safetensors'])
+supported_pt_extensions = set(['.ckpt', '.pt', '.bin', '.pth', '.safetensors'])

 folder_names_and_paths = {}