keep set_text

lucidrains · Apr 2, 2021 · b014b44 · b014b44
1 parent b01ce72
commit b014b44
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/big_sleep/big_sleep.py b/big_sleep/big_sleep.py
@@ -376,6 +376,9 @@ def __init__(
     def seed_suffix(self):
         return f'.{self.seed}' if self.append_seed and exists(self.seed) else ''
 
+    def set_text(self, text):
+        self.set_clip_encoding(text = text)
+
     def create_clip_encoding(self, text=None, img=None, encoding=None):
         self.text = text
         self.img = img

diff --git a/big_sleep/version.py b/big_sleep/version.py
@@ -1 +1 @@
-__version__ = '0.8.3'
+__version__ = '0.8.4'