fix tflearn#728

macesarr · Apr 27, 2017 · b9f48af · b9f48af
1 parent 257a827
commit b9f48af
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tflearn/models/generator.py b/tflearn/models/generator.py
@@ -217,7 +217,7 @@ def generate(self, seq_length, temperature=0.5, seq_seed=None,
             next_index = _sample(preds, temperature)
             next_char = self.rev_dic[next_index]
 
-            if type(sequence) == str:
+            if type(sequence) in [str, unicode]:
                 generated += next_char
                 sequence = sequence[1:] + next_char
                 whole_sequence += next_char