automatic installation update is added

This commit is contained in:
alpertunga-bile
2023-10-23 22:41:38 +03:00
parent 81847c2483
commit a41973d0c3
2 changed files with 32 additions and 0 deletions
+1
View File
@@ -1,5 +1,6 @@
#from .nodes.SegGPT import segGPTNode
from .nodes import SegformerNode,ImageCaptioningNode,ImageProcessingNode
NODE_CLASS_MAPPINGS = {
**SegformerNode.NODE_CLASS_MAPPINGS,
+31
View File
@@ -0,0 +1,31 @@
from importlib.util import find_spec
from os.path import exists
from subprocess import run
is_portable = True if exists("python_embeded") else False
def check_and_install_package(package_name: str, install_name: str = None) -> None:
if find_spec(package_name):
return
print(f"/_\ Installing {package_name}")
package = install_name if install_name else package_name
if is_portable:
command = f".\\python_embeded\\python.exe -s -m pip install {package}"
else:
command = f"pip install {package}"
process = run(command, shell=True, check=True, capture_output=True)
print("/_\ Checking packages")
check_and_install_package("transformers")
check_and_install_package("pillow")
check_and_install_package("matplotlib")
check_and_install_package("numpy")
check_and_install_package("scipy")
check_and_install_package("fastapi")
check_and_install_package("pytorch_lightning")
check_and_install_package("clip", "git+https://github.com/openai/CLIP.git")