diff --git a/backend/__init__.py b/backend/__init__.py new file mode 100644 index 0000000000..e69de29bb2 diff --git a/backend/modules/__init__.py b/backend/modules/__init__.py new file mode 100644 index 0000000000..e69de29bb2 diff --git a/setup.py b/setup.py index 101a1ca47e..3d656f503c 100644 --- a/setup.py +++ b/setup.py @@ -1,14 +1,36 @@ 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( name='invoke-ai', version='2.1.4', - description='InvokeAI text to image generation toolkit', + description='InvokeAI: A Stable Diffusion text to image generation toolkit', packages=find_packages(), install_requires=[ 'torch', 'numpy', 'tqdm', ], + cmdclass={ + 'develop': PostDevelopCommand, + 'install': PostInstallCommand, + }, )