extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2024-12-25 16:38:55 +01:00
Code Issues Packages Projects Releases Wiki Activity

Log the version numbers of only a few important modules

Browse Source
This commit is contained in:
cmdr2 2023-06-07 16:38:15 +05:30
parent 267c7b85ea
commit 48edce72a9
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/check_modules.py
View File

@ -26,6 +26,7 @@ modules_to_check = {
"pycloudflared": "0.2.0",
# "xformers": "0.0.16",
}
modules_to_log = ["torch", "torchvision", "sdkit", "stable-diffusion-sdkit"]
def version(module_name: str) -> str:
@ -90,7 +91,8 @@ def init():
traceback.print_exc()
fail(module_name)
print(f"{module_name}: {version(module_name)}")
if module_name in modules_to_log:
print(f"{module_name}: {version(module_name)}")
### utilities