Skip to content

Commit

Permalink
update caikit-tgis-backend version to access the updated proto fields
Browse files Browse the repository at this point in the history
Signed-off-by: waleedqk <waleedqk@ibm.com>
  • Loading branch information
waleedqk committed Jul 9, 2024
1 parent 30e4331 commit 2f75aa2
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
4 changes: 2 additions & 2 deletions caikit_nlp/modules/text_generation/text_generation_tgis.py
Original file line number Diff line number Diff line change
Expand Up @@ -238,7 +238,7 @@ def run(
generated_tokens: bool = True,
token_logprobs: bool = True,
token_ranks: bool = True,
include_stop_sequence: bool = True,
include_stop_sequence: Optional[bool] = True,
context: Optional[RuntimeServerContextType] = None,
) -> GeneratedTextResult:
f"""Run inference against the model running in TGIS.
Expand Down Expand Up @@ -299,7 +299,7 @@ def run_stream_out(
generated_tokens: bool = True,
token_logprobs: bool = True,
token_ranks: bool = True,
include_stop_sequence: bool = True,
include_stop_sequence: Optional[bool] = True,
context: Optional[RuntimeServerContextType] = None,
) -> Iterable[GeneratedTextStreamResult]:
f"""Run output stream inferencing for text generation module.
Expand Down
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@ classifiers=[
]
dependencies = [
"caikit[runtime-grpc,runtime-http]>=0.26.34,<0.27.0",
"caikit-tgis-backend>=0.1.34,<0.2.0",
"caikit-tgis-backend>=0.1.36,<0.2.0",
# TODO: loosen dependencies
"grpcio>=1.62.2", # explicitly pin grpc dependencies to a recent version to avoid pip backtracking
"grpcio-reflection>=1.62.2",
Expand Down

0 comments on commit 2f75aa2

Please sign in to comment.