From f443e3c6948b6e2a7d67d3e45ba9bc877e883716 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 8420d097..5048d34e 100644 --- a/ui/easydiffusion/model_manager.py +++ b/ui/easydiffusion/model_manager.py @@ -293,7 +293,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: