mirror of
https://github.com/easydiffusion/easydiffusion.git
synced 2025-06-30 22:50:51 +02:00
Ignore pickle scanning for .sft and .gguf files
This commit is contained in:
@ -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:
|
||||
|
Reference in New Issue
Block a user