From 90fdedcfc21d2bacd8461d5f5cd92ee55d407094 Mon Sep 17 00:00:00 2001
From: wassname <wassname@users.noreply.github.com>
Date: Wed, 25 Dec 2019 07:10:57 +0800
Subject: [PATCH] avoid loop

---
 gpt2generator.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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