nodes/alpertunga-bile-prompt-generator-comfyui
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

install and uninstall scripts are removed

Browse Source
This commit is contained in:
alpertunga-bile
2025-05-21 22:49:10 +03:00
parent be2c3e0406
commit 778cba50bb
2 changed files with 0 additions and 94 deletions
Download Patch File Download Diff File Expand all files Collapse all files
install.py
-45
View File
@@ -1,45 +0,0 @@
import subprocess
import sys
import platform
import generator.utility
common_packages = [
"transformers",
"accelerate",
"optimum",
"optimum[onnxruntime-gpu]",
"transformers[sentencepiece]",
"peft",
]
def install_packages(packages: list[str]) -> None:
for package in packages:
print(f"/_\\ Installing {package}")
command = f"{sys.executable} -m pip install {package}"
process = subprocess.run(command, shell=True, check=True, capture_output=True)
if process.returncode != 0:
print(f"{package} installation is failed\nError: {process.stdout}")
if __name__ == "__main__":
print(" Prompt Generator ComfyUI Node ".center(100, "-"))
print("/_\\ Thank you for installing Prompt Generator node")
packages_to_install = []
packages_to_install.extend(common_packages)
os_name = platform.system()
if os_name == "Linux":
packages_to_install.append("triton")
packages_to_install.append("bitsandbytes")
if os_name != "Linux" and generator.utility.check_torch_version_is_enough(2, 4):
packages_to_install.append("optimum-quanto")
install_packages(packages_to_install)
print("/_\\ Installation is compeleted successfully")
print("-" * 100)
uninstall.py
-49
View File
@@ -1,49 +0,0 @@
"""
Actually uninstalling packages is not suitable
because they maybe need by other packages too
"""
"""
import subprocess
import sys
import platform
import generator.utility
common_packages = [
"optimum",
"optimum[onnxruntime-gpu]",
"peft",
]
def uninstall_packages(packages: list[str]) -> None:
for package in packages:
print(f"/_\\ Uninstalling {package}")
command = f"{sys.executable} -m pip uninstall --yes {package}"
process = subprocess.run(command, shell=True, check=True, capture_output=True)
if process.returncode != 0:
print(f"{package} uninstallation is failed\nError: {process.stdout}")
"""
if __name__ == "__main__":
print(" Prompt Generator ComfyUI Node ".center(100, "-"))
print(
"/_\\ If you have encountered issue(s) please feel free to open issue on repository's Github page"
)
"""
packages_to_uninstall = []
packages_to_uninstall.extend(common_packages)
os_name = platform.system()
if os_name != "Linux" and generator.utility.check_torch_version_is_enough(2, 4):
packages_to_uninstall.append("optimum-quanto")
uninstall_packages(packages_to_uninstall)
"""
print("/_\\ Uninstallation is compeleted successfully")
print("-" * 100)