Ignore pickle scanning for .sft and .gguf files

This commit is contained in:
cmdr2
2025-06-27 16:43:41 +05:30
parent 3a07523d08
commit f443e3c694

View File

@ -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: