diff --git a/gpt2generator.py b/gpt2generator.py index 59fe29d..52fd2a4 100644 --- a/gpt2generator.py +++ b/gpt2generator.py @@ -224,6 +224,6 @@ class GPT2Generator: result = text result = self.result_replace(result) if len(result) == 0: - return self.generate(prompt) - + logger.warn("Model generated empty text %s.", result) + # return self.generate(prompt) # Woah recursion! return result