rebase and update FinishReason

neuralmagic · Sep 15, 2023 · 6a79bca · 6a79bca
1 parent bf1aa90
commit 6a79bca
Showing 1 changed file with 3 additions and 3 deletions.
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):