okket commited on
Commit
13eee04
1 Parent(s): 4f68cb1

Add prompt format hint

Browse files

correct format according to reddit:

https://www.reddit.com/r/LocalLLaMA/comments/18ckls0/mambachat_a_chat_llm_based_on_state_space_models/

Files changed (1) hide show
  1. README.md +10 -1
README.md CHANGED
@@ -6,4 +6,13 @@ Mamba-Chat is the first chat language model based on a state-space model archite
6
 
7
  The model is a fine-tune of Albert Gu's and Tri Dao's model [Mamba-2.8B](https://github.com/state-spaces/mamba) from their paper *Mamba: Linear-Time Sequence Modeling with Selective State Spaces*.
8
 
9
- Check our our [Github repository](https://github.com/havenhq/mamba-chat/tree/main) for training and inference code.
 
 
 
 
 
 
 
 
 
 
6
 
7
  The model is a fine-tune of Albert Gu's and Tri Dao's model [Mamba-2.8B](https://github.com/state-spaces/mamba) from their paper *Mamba: Linear-Time Sequence Modeling with Selective State Spaces*.
8
 
9
+ Check our our [Github repository](https://github.com/havenhq/mamba-chat/tree/main) for training and inference code.
10
+
11
+ The prompt format is the zephyr format:
12
+
13
+ ```
14
+ <|user|> {user_message}
15
+ <|assistant|> {assistant_message}
16
+ <|user|> {user_message}
17
+ <|assistant|> {assistant_message}
18
+ ```