[update]edit main

main.py

@@ -127,8 +127,6 @@ def main():
 
         output = ""
         for output_ in streamer:
-            output_ = output_.tolist()[0][len(input_ids[0]):]
-            output_ = tokenizer.decode(output_)
             output_ = output_.strip().replace(tokenizer.eos_token, "").strip()
 
             output += output_