wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-28 23:30:16 +08:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

remove bitsandbytes, rembg deps

Browse Source
This commit is contained in:
Benjamin Berman 2023-08-21 12:36:42 -07:00
parent 8a3331609e
commit 9897c9c9bd
2 changed files with 1 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
requirements.txt
View File

@ -25,8 +25,5 @@ importlib_resources
Pillow Pillow
scipy scipy
tqdm tqdm
diffusers>=0.16.1
protobuf==3.20.3 protobuf==3.20.3
rembg psutil
psutil
bitsandbytes>=0.40.1

14
setup.py
View File

@ -49,16 +49,6 @@ cpu_torch_index = "https://download.pytorch.org/whl/cpu"
# xformers not required for new torch # xformers not required for new torch
"""
Packages that should have a specific option set when a GPU accelerator is present
"""
gpu_accelerated_packages = {"rembg": "rembg[gpu]"}
"""
The URL to the bitsandbytes package to use on Windows
"""
bitsandbytes_windows = "https://github.com/jllllll/bitsandbytes-windows-webui/releases/download/wheels/bitsandbytes-0.40.1.post1-py3-none-win_amd64.whl"
""" """
Indicates if we're installing an editable (develop) mode package Indicates if we're installing an editable (develop) mode package
""" """
@ -157,10 +147,6 @@ def dependencies() -> [str]:
requirement = InstallRequirement(Requirement(package), comes_from=f"{package_name}=={version}") requirement = InstallRequirement(Requirement(package), comes_from=f"{package_name}=={version}")
candidate = finder.find_best_candidate(requirement.name, requirement.specifier) candidate = finder.find_best_candidate(requirement.name, requirement.specifier)
if candidate.best_candidate is not None: if candidate.best_candidate is not None:
if requirement.name == "bitsandbytes" and platform.system().lower() == 'windows':
_dependencies[i] = f"{requirement.name} @ {bitsandbytes_windows}"
if gpu_accelerated and requirement.name in gpu_accelerated_packages:
_dependencies[i] = gpu_accelerated_packages[requirement.name]
if any([url in candidate.best_candidate.link.url for url in _alternative_indices]): if any([url in candidate.best_candidate.link.url for url in _alternative_indices]):
_dependencies[i] = f"{requirement.name} @ {candidate.best_candidate.link.url}" _dependencies[i] = f"{requirement.name} @ {candidate.best_candidate.link.url}"
return _dependencies return _dependencies