1580158
/
PBRemTools
mirror of https://github.com/mattyamonaca/PBRemTools.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
install.py
View File

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