Spaces:

aka7774
/

trllm

Sleeping

aka7774 commited on about 1 month ago

Commit

7b90ef7

•

1 Parent(s): c3dba91

Update fn.py

Files changed (1) hide show

fn.py CHANGED Viewed

@@ -207,7 +207,7 @@ def chat(message, history = [], instruction = None, args = {}):
                 yield model_output
     outputs = model.generate(**generate_kwargs)
-    content = tokenizer.decode(outputs, skip_special_tokens=True)
     return content
 def apply_template(messages):

                 yield model_output
     outputs = model.generate(**generate_kwargs)
+    content = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return content
 def apply_template(messages):