clibrain
/

mamba-2.8b-instruct-openhermes

Text Generation

Inference Endpoints

Model card Files Files and versions Community

mrm8488 commited on Dec 31, 2023

Commit

6212a3f

•

1 Parent(s): 605742c

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

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