rebase

neuralmagic · Sep 18, 2023 · a69abf9 · a69abf9
1 parent f002e13
commit a69abf9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/deepsparse/transformers/pipelines/text_generation.py b/src/deepsparse/transformers/pipelines/text_generation.py
@@ -57,6 +57,7 @@ class FinishReason(Enum):
     STOP = "stop"
     LENGTH = "length"
     TIME = "time"
+    CALLBACK = "callback"
 
 
 class TextGenerationInput(BaseModel):