add performance graph to readme

2022-07-11 08:39:48 -04:00
parent bd9287cec1
commit 2ab4365059
3 changed files with 5 additions and 1 deletions
--- a/min_dalle/min_dalle.py
+++ b/min_dalle/min_dalle.py
@@ -186,7 +186,7 @@ class MinDalle:
        tokens = self.tokenizer.tokenize(text, is_verbose=is_verbose)
        if len(tokens) > self.text_token_count: 
            tokens = tokens[:self.text_token_count]
-        if is_verbose: print("text tokens", tokens)
+        if is_verbose: print("{} text tokens".format(len(tokens)), tokens)
        text_tokens = numpy.ones((2, 64), dtype=numpy.int32)
        text_tokens[0, :2] = [tokens[0], tokens[-1]]
        text_tokens[1, :len(tokens)] = tokens