From 95245425ebc7fa76f2ec61cdd96689f25c5df872 Mon Sep 17 00:00:00 2001
From: Qing <cwq1913@gmail.com>
Date: Sun, 16 Apr 2023 09:59:35 +0800
Subject: [PATCH] add more package version check

---
 lama_cleaner/runtime.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/lama_cleaner/runtime.py b/lama_cleaner/runtime.py
index 319894f..d8cc2e0 100644
--- 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: