rebase and update FinishReason

dsikka · dsikka · commit 6a79bcadc7da · 2023-09-15T13:38:56.000-04:00
diff --git a/src/deepsparse/transformers/pipelines/text_generation.py b/src/deepsparse/transformers/pipelines/text_generation.py
@@ -55,9 +55,9 @@
 
 
 class FinishReason(Enum):
-    STOP = "stop token generated."
-    LENGTH = "max token generation reached."
-    TIME = "max time reached."
+    STOP = "stop"
+    LENGTH = "length"
+    TIME = "time"
 
 
 class TextGenerationInput(BaseModel):
