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 13 Packages Projects Releases Wiki Activity

Investigate better support for PyPi

Browse Source
This commit is contained in:
doctorpangloss 2024-06-13 13:34:48 -07:00
parent f6388683e0
commit 3cc3173625
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
setup.py
View File

@ -117,8 +117,10 @@ def _is_linux_arm64():
return os_name == 'Linux' and architecture == 'aarch64' return os_name == 'Linux' and architecture == 'aarch64'
def dependencies(force_nightly: bool = False) -> List[str]: def dependencies(for_pypi=False, force_nightly: bool = False) -> List[str]:
_dependencies = open(os.path.join(os.path.dirname(__file__), "requirements.txt")).readlines() _dependencies = open(os.path.join(os.path.dirname(__file__), "requirements.txt")).readlines()
if for_pypi:
return [dep for dep in _dependencies if dep not in {"torch", "torchvision"} and "@" not in dep]
# If we're installing with no build isolation, we can check if torch is already installed in the environment, and if # If we're installing with no build isolation, we can check if torch is already installed in the environment, and if
# so, go ahead and use the version that is already installed. # so, go ahead and use the version that is already installed.
existing_torch: Optional[str] existing_torch: Optional[str]
@ -202,7 +204,7 @@ setup(
python_requires=">=3.10,<3.13", python_requires=">=3.10,<3.13",
packages=find_packages(exclude=["tests"] + [] if is_editable else ['custom_nodes']), packages=find_packages(exclude=["tests"] + [] if is_editable else ['custom_nodes']),
include_package_data=True, include_package_data=True,
install_requires=dependencies(), install_requires=dependencies(for_pypi=False),
setup_requires=["pip", "wheel"], setup_requires=["pip", "wheel"],
entry_points={ entry_points={
'console_scripts': [ 'console_scripts': [