create the eventual dream cli, like bigsleep library

This commit is contained in:
Phil Wang
2022-04-12 10:04:17 -07:00
parent b93ad8b7a2
commit 2ab042b862
2 changed files with 6 additions and 0 deletions

View File

@@ -2,3 +2,8 @@ import click
def main(): def main():
pass pass
@click.command()
@click.argument('text')
def dream(text):
return image

View File

@@ -7,6 +7,7 @@ setup(
entry_points={ entry_points={
'console_scripts': [ 'console_scripts': [
'dalle2_pytorch = dalle2_pytorch.cli:main', 'dalle2_pytorch = dalle2_pytorch.cli:main',
'dream = dalle2_pytorch.cli:dream'
], ],
}, },
version = '0.0.2', version = '0.0.2',