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"