From b5394dce8e85a0fa519cfdffd6c919dcdc3a9bc4 Mon Sep 17 00:00:00 2001 From: Frank Liu Date: Fri, 2 Jun 2023 09:28:57 -0700 Subject: [PATCH] [python] Fixes typo in transformers-neuronx.py --- engines/python/setup/djl_python/transformers-neuronx.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/engines/python/setup/djl_python/transformers-neuronx.py b/engines/python/setup/djl_python/transformers-neuronx.py index b92b9c05a..eafafbe48 100644 --- a/engines/python/setup/djl_python/transformers-neuronx.py +++ b/engines/python/setup/djl_python/transformers-neuronx.py @@ -185,7 +185,7 @@ def infer(self, inputs): model_kwargs = {} if self.enable_streaming: - stream_generator, c_type = StreamingUtils.get_stream_generator( + stream_generator, ctype = StreamingUtils.get_stream_generator( "transformers-neuronx") outputs.add_property("content-type", ctype) model_kwargs["engine"] = "transformers-neuronx"