mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
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,
|
||
|
|
)
|