kamp/IOPaint
1
0
Fork 0
Code Issues Pull Requests Releases 1 Activity

add more package version check

Browse Source
This commit is contained in:
Qing 2023-04-16 09:59:35 +08:00
parent d3f6f69af6
commit 95245425eb
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
lama_cleaner/runtime.py
View File

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