add more package version check

2023-04-16 09:59:35 +08:00 · 2023-04-16 09:59:35 +08:00 · 95245425eb
commit 95245425eb
parent d3f6f69af6
1 changed files with 5 additions and 1 deletions
--- a/lama_cleaner/runtime.py
+++ b/lama_cleaner/runtime.py
@ -16,13 +16,17 @@ _package_versions = {}

 _CANDIDATES = [
    "torch",
+    "torchvision",
    "Pillow",
    "diffusers",
    "transformers",
    "opencv-python",
    "xformers",
    "accelerate",
-    "lama-cleaner"
+    "lama-cleaner",
+    "rembg",
+    "realesrgan",
+    "gfpgan",
 ]
 # Check once at runtime
 for name in _CANDIDATES: