From 48edce72a96d08f646221f79b19d26a6842e20d6 Mon Sep 17 00:00:00 2001
From: cmdr2 <secondary.cmdr2@gmail.com>
Date: Wed, 7 Jun 2023 16:38:15 +0530
Subject: [PATCH] Log the version numbers of only a few important modules

---
 scripts/check_modules.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/scripts/check_modules.py b/scripts/check_modules.py
index 5264ec08..dc3733e9 100644
--- a/scripts/check_modules.py
+++ b/scripts/check_modules.py
@@ -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