holm/easydiffusion
1
0
Fork 0
forked from extern/easydiffusion
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Picklescanner is mandatory

Browse Source
This commit is contained in:
cmdr2 2022-11-30 13:38:22 +05:30
parent efd9a22bb5
commit c64bcd23d3
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ui/server.py
View File

@ -7,6 +7,7 @@ import traceback
import sys import sys
import os import os
import picklescan.scanner
import rich import rich
SD_DIR = os.getcwd() SD_DIR = os.getcwd()
@ -234,7 +235,6 @@ async def setAppConfig(req : SetAppConfigRequest):
def is_malicious_model(file_path): def is_malicious_model(file_path):
try: try:
import picklescan.scanner
scan_result = picklescan.scanner.scan_file_path(file_path) scan_result = picklescan.scanner.scan_file_path(file_path)
if scan_result.issues_count > 0 or scan_result.infected_files > 0: if scan_result.issues_count > 0 or scan_result.infected_files > 0:
rich.print(":warning: [bold red]Scan %s: %d scanned, %d issue, %d infected.[/bold red]" % (file_path, scan_result.scanned_files, scan_result.issues_count, scan_result.infected_files)) rich.print(":warning: [bold red]Scan %s: %d scanned, %d issue, %d infected.[/bold red]" % (file_path, scan_result.scanned_files, scan_result.issues_count, scan_result.infected_files))