clibrain
/

mamba-2.8b-ft-synthia-v1.3

Model card Files Files and versions Community

mrm8488 commited on Dec 31, 2023

Commit

e47d9ad

•

1 Parent(s): f078fb6

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -70,7 +70,7 @@ out = model.generate(
 decoded = tokenizer.batch_decode(out)
 assistant_message = (
-    decoded[0].split("<|assistant|>\n")[-1].replace(eos, "")
 )
 print(assistant_message)

 decoded = tokenizer.batch_decode(out)
 assistant_message = (
+    decoded[0].split("<|assistant|>\n")[-1].replace(eos_token, "")
 )
 print(assistant_message)