Writer
/

camel-5b-hf

Text Generation

text-generation-inference

Model card Files Files and versions

kiranr commited on Apr 10, 2023

Commit

c59532d

·

1 Parent(s): e2c44ff

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -54,7 +54,7 @@ output_ids = model.generate(
     **model_inputs,
     max_length=100,
 )
-output_text = tokenizer.batch_decode(output_ids, skip_special_tokens=True)
 clean_output = output_text.split("### Response:")[1].strip()
 print(clean_output)

     **model_inputs,
     max_length=100,
 )
+output_text = tokenizer.batch_decode(output_ids, skip_special_tokens=True)[0]
 clean_output = output_text.split("### Response:")[1].strip()
 print(clean_output)