Safetensors isn't optional anymore.

679bd284 · comfyanonymous · eb4bd771 · 679bd284
Commit 679bd284 authored May 26, 2023 by comfyanonymous
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 9 deletions

folder_paths.py folder_paths.py +2 -9

No files found.
--- a/folder_paths.py
+++ b/folder_paths.py
 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 = {}