diff --git a/ui/easydiffusion/model_manager.py b/ui/easydiffusion/model_manager.py index 795eea73..116edf33 100644 --- a/ui/easydiffusion/model_manager.py +++ b/ui/easydiffusion/model_manager.py @@ -158,6 +158,8 @@ def make_model_folders(): def is_malicious_model(file_path): try: + if file_path.endswith(".safetensors"): + return False scan_result = scan_model(file_path) if scan_result.issues_count > 0 or scan_result.infected_files > 0: log.warn(