From 2ab042b862f29f32b2c31c4fd4be7dc2d334ea60 Mon Sep 17 00:00:00 2001
From: Phil Wang <lucidrains@gmail.com>
Date: Tue, 12 Apr 2022 10:04:17 -0700
Subject: [PATCH] create the eventual dream cli, like bigsleep library

---
 dalle2_pytorch/cli.py | 5 +++++
 setup.py              | 1 +
 2 files changed, 6 insertions(+)

diff --git a/dalle2_pytorch/cli.py b/dalle2_pytorch/cli.py
index 5b60af9..ed87e8b 100644
--- a/dalle2_pytorch/cli.py
+++ b/dalle2_pytorch/cli.py
@@ -2,3 +2,8 @@ import click
 
 def main():
     pass
+
+@click.command()
+@click.argument('text')
+def dream(text):
+    return image
diff --git a/setup.py b/setup.py
index 05216f8..1672c3a 100644
--- a/setup.py
+++ b/setup.py
@@ -7,6 +7,7 @@ setup(
   entry_points={
     'console_scripts': [
       'dalle2_pytorch = dalle2_pytorch.cli:main',
+      'dream = dalle2_pytorch.cli:dream'
     ],
   },
   version = '0.0.2',