Add installation of timm required to run lama-cleaner

main
Uminosachi 2024-05-07 18:46:44 +09:00
parent a980f40a29
commit ed336d45b0
1 changed files with 6 additions and 0 deletions

View File

@ -66,6 +66,12 @@ if not launch.is_installed("pydantic"):
except Exception:
print("Can't install pydantic. Please follow the readme to install manually")
if not launch.is_installed("timm"):
try:
launch.run_pip("install timm", "requirements for timm")
except Exception:
print("Can't install timm. Please follow the readme to install manually")
if not launch.is_installed("ultralytics"):
try:
launch.run_pip("install ultralytics", "requirements for ultralytics")