clibrain
/

mamba-2.8b-chat-no_robots

Text Generation

Model card Files Files and versions

Metrics Training metrics Community

mrm8488 commited on Dec 31, 2023

Commit

b4a1811

·

1 Parent(s): 744cfe1

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -95,7 +95,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)