mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
9997fde144
This allows reliable distribution of the initial 'configs' directory with the Python package, and enables the configuration script to be running from anywhere, as long as the virtual environment is available on the sys.path
101 lines
3.3 KiB
Python
101 lines
3.3 KiB
Python
import sys
|
|
import os
|
|
import re
|
|
from setuptools import setup, find_packages
|
|
|
|
def list_files(directory):
|
|
listing = list()
|
|
for root, dirs, files in os.walk(directory,topdown=False):
|
|
pair = (root,[os.path.join(root,f) for f in files])
|
|
listing.append(pair)
|
|
return listing
|
|
|
|
|
|
def get_version()->str:
|
|
from ldm.invoke import __version__ as version
|
|
return version
|
|
|
|
# The canonical version number is stored in the file ldm/invoke/_version.py
|
|
VERSION = get_version()
|
|
DESCRIPTION = ('An implementation of Stable Diffusion which provides various new features'
|
|
' and options to aid the image generation process')
|
|
LONG_DESCRIPTION = ('This version of Stable Diffusion features a slick WebGUI, an'
|
|
' interactive command-line script that combines text2img and img2img'
|
|
' functionality in a "dream bot" style interface, and multiple features'
|
|
' and other enhancements.')
|
|
HOMEPAGE = 'https://github.com/invoke-ai/InvokeAI'
|
|
FRONTEND_FILES = list_files('frontend/dist')
|
|
FRONTEND_FILES.append(('assets',['assets/caution.png']))
|
|
print(FRONTEND_FILES)
|
|
|
|
REQUIREMENTS=[
|
|
'accelerate',
|
|
'albumentations',
|
|
'diffusers',
|
|
'eventlet',
|
|
'flask_cors',
|
|
'flask_socketio',
|
|
'flaskwebgui',
|
|
'getpass_asterisk',
|
|
'imageio-ffmpeg',
|
|
'pyreadline3',
|
|
'realesrgan',
|
|
'send2trash',
|
|
'streamlit',
|
|
'taming-transformers-rom1504',
|
|
'test-tube',
|
|
'torch-fidelity',
|
|
'torch',
|
|
'torchvision',
|
|
'transformers',
|
|
'picklescan',
|
|
'clip',
|
|
'clipseg',
|
|
'gfpgan',
|
|
'k-diffusion',
|
|
'pypatchmatch',
|
|
]
|
|
|
|
setup(
|
|
name='InvokeAI',
|
|
version=VERSION,
|
|
description=DESCRIPTION,
|
|
long_description=LONG_DESCRIPTION,
|
|
author='The InvokeAI Project',
|
|
author_email='lincoln.stein@gmail.com',
|
|
url=HOMEPAGE,
|
|
license='MIT',
|
|
packages=find_packages(exclude=['tests.*']),
|
|
install_requires=REQUIREMENTS,
|
|
dependency_links=['https://download.pytorch.org/whl/torch_stable.html'],
|
|
python_requires='>=3.9, <3.11',
|
|
classifiers=[
|
|
'Development Status :: 4 - Beta',
|
|
'Environment :: GPU',
|
|
'Environment :: GPU :: NVIDIA CUDA',
|
|
'Environment :: MacOS X',
|
|
'Intended Audience :: End Users/Desktop',
|
|
'Intended Audience :: Developers',
|
|
'License :: OSI Approved :: MIT License',
|
|
'Operating System :: POSIX :: Linux',
|
|
'Operating System :: MacOS',
|
|
'Operating System :: Microsoft :: Windows',
|
|
'Programming Language :: Python :: 3 :: Only,'
|
|
'Programming Language :: Python :: 3.8',
|
|
'Programming Language :: Python :: 3.9',
|
|
'Programming Language :: Python :: 3.10',
|
|
'Topic :: Artistic Software',
|
|
'Topic :: Internet :: WWW/HTTP :: WSGI :: Application',
|
|
'Topic :: Internet :: WWW/HTTP :: WSGI :: Server',
|
|
'Topic :: Multimedia :: Graphics',
|
|
'Topic :: Scientific/Engineering :: Artificial Intelligence',
|
|
'Topic :: Scientific/Engineering :: Image Processing',
|
|
],
|
|
scripts = ['scripts/invoke.py','scripts/configure_invokeai.py', 'scripts/sd-metadata.py',
|
|
'scripts/images2prompt.py','scripts/merge_embeddings.py',
|
|
'scripts/textual_inversion_fe.py','scripts/textual_inversion.py'
|
|
],
|
|
data_files=FRONTEND_FILES,
|
|
include_package_data=True,
|
|
)
|