add TODO statements

neuralmagic · Sep 14, 2023 · 115a7d7 · 115a7d7
1 parent dff4c84
commit 115a7d7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/deepsparse/transformers/pipelines/text_generation.py b/src/deepsparse/transformers/pipelines/text_generation.py
@@ -146,6 +146,7 @@ class GeneratedText(BaseModel):
     )
 
 
+# TODO: Pydantic aliases allow assignment but not reference. Still need to update.
 class TextGenerationOutput(BaseModel):
     created: datetime.datetime = Field(description="Time of inference creation.")
     prompts: Union[str, List[str]] = Field(
@@ -609,6 +610,7 @@ def engine_forward(
                         finished_reason.append(FinishReason.STOP)
                         break
 
+                    # TODO: Add any generic callback reason?
                     if callback is not None and callback(token) is False:
                         _LOGGER.debug(
                             "callback %s returned False, stopping generation."