fix save best after set_text is called

lucidrains · Apr 3, 2021 · 056e6fd · 056e6fd
1 parent b014b44
commit 056e6fd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/big_sleep/big_sleep.py b/big_sleep/big_sleep.py
@@ -421,6 +421,7 @@ def encode_max_and_min(self, text, img=None, encoding=None, text_min=""):
             self.encode_multiple_phrases(text_min, img=img, encoding=encoding, text_type="min")
 
     def set_clip_encoding(self, text=None, img=None, encoding=None, text_min=""):
+        self.current_best_score = 0
         self.text = text
         self.text_min = text_min
 

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