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.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,
|
||||
},
|
||||
)
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user