extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2024-11-22 08:13:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

Log errors during module and model initialization

Browse Source
This commit is contained in:
cmdr2 2023-04-19 16:20:08 +05:30
parent 7c75a61700
commit 35c75115de
2 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/check_models.py
View File

@ -6,6 +6,7 @@ from sdkit.utils import hash_file_quick
import os
import shutil
from glob import glob
import traceback
models_base_dir = os.path.abspath(os.path.join("..", "models"))
@ -55,6 +56,7 @@ def init():
try:
download_if_necessary(model_type, model["file_name"], model["model_id"])
except:
traceback.print_exc()
fail(model_type)
print(model_type, "model(s) found.")

2
scripts/check_modules.py
View File

@ -11,6 +11,7 @@ a custom index URL depending on the platform.
import os
from importlib.metadata import version as pkg_version
import platform
import traceback
os_name = platform.system()
@ -58,6 +59,7 @@ def init():
try:
install(module_name, latest_version)
except:
traceback.print_exc()
fail(module_name)
print(f"{module_name}: {version(module_name)}")