nodes/LevelPixel-ComfyUI-LevelPixel
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix uninstall onnxruntime

Browse Source
This commit is contained in:
Level Pixel Dev
2025-05-28 07:13:47 +06:00
parent 32861175aa
commit 574d39b48f
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
install_init.py
+5 -2
View File
@@ -186,11 +186,14 @@ def install_onnxruntime_and_rembg():
install("onnxruntime-gpu>=1.22")
else:
if is_installed('onnxruntime-gpu'):
uninstall("onnxruntime-gpu")
#uninstall("onnxruntime-gpu")
print(f"LP >>> Your python has the 'onnxruntime-gpu' library installed, but you don't have a GPU.")
print(f"LP >>> Solution: If other node packages do not use the 'onnxruntime-gpu' library, then remove the 'onnxruntime-gpu' library for your python.")
print(f"LP >>> Close ComfyUI and run the script at .\\ComfyUI\\custom_nodes\\ComfyUI-LevelPixel\\scripts\\remove_onnxruntime.bat")
if not is_installed('onnxruntime'):
install("onnxruntime>=1.22")
rembg_pkg = 'rembg[gpu]' if gpu else 'rembg[cpu]'
rembg_pkg = 'rembg[gpu]' if gpu else 'rembg[cpu]'
if not is_installed('rembg', '2.0.66'):
install(f'{rembg_pkg}>={'2.0.66'}')