From 679bd2845af8e22b2802cf326b99b40a26ba7811 Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Fri, 26 May 2023 21:46:11 -0400
Subject: [PATCH] Safetensors isn't optional anymore.

---
 folder_paths.py | 11 ++---------
 1 file changed, 2 insertions(+), 9 deletions(-)

diff --git a/folder_paths.py b/folder_paths.py
index 28f11782..20b461c9 100644
--- 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 = {}