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: