avoid loop

2026-09-09 11:13:26 +08:00 · 2019-12-25 07:10:57 +08:00
parent 57432290ea
commit 90fdedcfc2
1 changed files with 2 additions and 2 deletions
@@ -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