From 947a1bfdecf201f78bc96c469b38fee9fbab2831 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 25 Feb 2023 11:24:28 -0600 Subject: [PATCH] fix variable names --- api/onnx_web/transformers.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/api/onnx_web/transformers.py b/api/onnx_web/transformers.py index 31c066e4..f7a70693 100644 --- a/api/onnx_web/transformers.py +++ b/api/onnx_web/transformers.py @@ -27,14 +27,14 @@ def run_txt2txt_pipeline( input_ids = tokenizer.encode(params.prompt, return_tensors="pt").to( device.torch_device() ) - output = model.generate( + results = model.generate( input_ids, do_sample=True, max_length=tokens, temperature=0.8, ) - result = tokenizer.decode(output[0], skip_special_tokens=True) + result_text = tokenizer.decode(results[0], skip_special_tokens=True) - print("Server says: %s" % result) + print("Server says: %s" % result_text) logger.info("finished txt2txt job: %s", output)