From 83b8028e0aeba076eaaef3dc6f52f74aa37a94a2 Mon Sep 17 00:00:00 2001 From: cmdr2 Date: Fri, 27 Jun 2025 16:43:41 +0530 Subject: [PATCH] Ignore pickle scanning for .sft and .gguf files --- ui/easydiffusion/model_manager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ui/easydiffusion/model_manager.py b/ui/easydiffusion/model_manager.py index 39e5ec7a..45b38b96 100644 --- a/ui/easydiffusion/model_manager.py +++ b/ui/easydiffusion/model_manager.py @@ -325,7 +325,7 @@ def make_model_folders(): def is_malicious_model(file_path): try: - if file_path.endswith(".safetensors"): + if file_path.endswith((".safetensors", ".sft", ".gguf")): return False scan_result = scan_model(file_path) if scan_result.issues_count > 0 or scan_result.infected_files > 0: