From 69683fec94ee8b25ef25b1fbd505adab3aeb53f3 Mon Sep 17 00:00:00 2001
From: Brett Kuprel <brkuprel@gmail.com>
Date: Fri, 1 Jul 2022 16:02:05 -0400
Subject: [PATCH] simplify import

---
 image_from_text.py    | 6 +++---
 min_dalle.ipynb       | 2 +-
 min_dalle/__init__.py | 1 +
 3 files changed, 5 insertions(+), 4 deletions(-)
 create mode 100644 min_dalle/__init__.py

diff --git a/image_from_text.py b/image_from_text.py
index 183dc5e..5ede49f 100644
--- a/image_from_text.py
+++ b/image_from_text.py
@@ -2,14 +2,14 @@ import argparse
 import os
 from PIL import Image
 
-from min_dalle.min_dalle_torch import MinDalleTorch
+from min_dalle import MinDalleTorch
 
 parser = argparse.ArgumentParser()
 parser.add_argument('--mega', action='store_true')
 parser.add_argument('--no-mega', dest='mega', action='store_false')
 parser.set_defaults(mega=False)
-parser.add_argument('--text', type=str, default='cat')
-parser.add_argument('--seed', type=int, default=0)
+parser.add_argument('--text', type=str, default='alien life')
+parser.add_argument('--seed', type=int, default=7)
 parser.add_argument('--image_path', type=str, default='generated')
 parser.add_argument('--token_count', type=int, default=256) # for debugging
 
diff --git a/min_dalle.ipynb b/min_dalle.ipynb
index c0bab7e..786106a 100644
--- a/min_dalle.ipynb
+++ b/min_dalle.ipynb
@@ -134,7 +134,7 @@
       "source": [
         "import os\n",
         "os.chdir('/content/min-dalle')\n",
-        "from min_dalle.min_dalle_torch import MinDalleTorch\n",
+        "from min_dalle import MinDalleTorch\n",
         "\n",
         "model = MinDalleTorch(is_mega=True, is_reusable=True)"
       ]
diff --git a/min_dalle/__init__.py b/min_dalle/__init__.py
new file mode 100644
index 0000000..9807201
--- /dev/null
+++ b/min_dalle/__init__.py
@@ -0,0 +1 @@
+from .min_dalle_torch import MinDalleTorch
\ No newline at end of file