mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
add module __init__ files for backend
This commit is contained in:
parent
9200b26f21
commit
fdb16000ab
0
backend/__init__.py
Normal file
0
backend/__init__.py
Normal file
0
backend/modules/__init__.py
Normal file
0
backend/modules/__init__.py
Normal file
24
setup.py
24
setup.py
@ -1,14 +1,36 @@
|
|||||||
from setuptools import setup, find_packages
|
from setuptools import setup, find_packages
|
||||||
|
from setuptools.command.develop import develop
|
||||||
|
from setuptools.command.install import install
|
||||||
|
|
||||||
|
class PostDevelopCommand(develop):
|
||||||
|
"""Post-installation for development mode."""
|
||||||
|
def run(self):
|
||||||
|
develop.run(self)
|
||||||
|
print('Will now try loading a module (develop)')
|
||||||
|
import ldm.generate
|
||||||
|
print('ldm.generate loaded ok')
|
||||||
|
|
||||||
|
class PostInstallCommand(install):
|
||||||
|
"""Post-installation for installation mode."""
|
||||||
|
def run(self):
|
||||||
|
install.run(self)
|
||||||
|
print('Will now try loading a module (install)')
|
||||||
|
import ldm.generate
|
||||||
|
print('ldm.generate loaded ok')
|
||||||
|
|
||||||
setup(
|
setup(
|
||||||
name='invoke-ai',
|
name='invoke-ai',
|
||||||
version='2.1.4',
|
version='2.1.4',
|
||||||
description='InvokeAI text to image generation toolkit',
|
description='InvokeAI: A Stable Diffusion text to image generation toolkit',
|
||||||
packages=find_packages(),
|
packages=find_packages(),
|
||||||
install_requires=[
|
install_requires=[
|
||||||
'torch',
|
'torch',
|
||||||
'numpy',
|
'numpy',
|
||||||
'tqdm',
|
'tqdm',
|
||||||
],
|
],
|
||||||
|
cmdclass={
|
||||||
|
'develop': PostDevelopCommand,
|
||||||
|
'install': PostInstallCommand,
|
||||||
|
},
|
||||||
)
|
)
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user