Update install.py

Fixes library names so launch.is_installed() properly checks for installed packages
pull/38/head
predprey2 2023-05-23 05:46:23 +08:00 committed by GitHub
parent b179be6192
commit c83dd8c53a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 5 deletions

View File

@ -2,12 +2,12 @@ import launch
packages = {
"onnx": "onnx",
"onnxruntime-gpu": "onnxruntime-gpu==1.14.0",
"opencv-python": "opencv-python",
"onnxruntime": "onnxruntime-gpu==1.14.0",
"cv2": "opencv-python",
"numpy": "numpy",
"Pillow": "Pillow",
"segmentation-refinement": "segmentation-refinement",
"scikit-learn": "scikit-learn",
"PIL": "Pillow",
"segmentation_refinement": "segmentation-refinement",
"sklearn": "scikit-learn",
"clip": "clip",
}