forked from city96/ComfyUI_ExtraModels
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsetup.py
32 lines (28 loc) · 908 Bytes
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
from os import path
from platform import system
from urllib import parse
from setuptools import setup, find_packages
setup_dir = path.dirname(path.abspath(__file__))
if system() != 'Windows' and ' ' in setup_dir:
setup_dir = parse.quote(setup_dir)
our_requirements = open("requirements.txt").readlines()
our_requirements = [req.replace("{root:uri}", f"file://{setup_dir}") for req in our_requirements]
packages = find_packages()
setup(
name="comfyui_extra_models",
version="0.0.1",
packages=packages,
install_requires=our_requirements,
author='',
author_email='',
description='',
include_package_data=True,
entry_points={
'comfyui.custom_nodes': [
'comfyui_extra_models = comfyui_extra_models.nodes',
],
},
package_data={
package_or_module: ['*.json', 'spiece.model', "*.txt"] for package_or_module in packages
},
)