Condense/contextualize description

README.md

@@ -5,12 +5,11 @@ license: llama2
 ## Description
 
 This model is intended to be used as an accelerator for [llama 13B (chat)](https://huggingface.co/meta-llama/Llama-2-13b-chat-hf) and takes inspiration 
-from the Medusa architecture and modifies the MLP into a multi-stage MLP, where each stage predicts 
-a single token in the draft. Each stage takes as input both a state vector and sampled token embedding 
-from the prior stage (the base model can be considered stage 0). The inputs are projected and passed 
-through a LayerNorm/GeLU activation, forming a new state vector. This state vector is used to predict 
-the next draft token, which, with the new state vector, acts as input for the next stage of prediction. 
-We sample multiple tokens at each stage, and emit a tree of candidate suffixes to evaluate in parallel.
+from the Medusa speculative decoding architecture. This accelerator modifies the MLP into a multi-stage MLP, where each stage predicts 
+a single token in the draft based on both a state vector and sampled token
+from the prior stage (the base model can be considered stage 0).
+The state vector from the base model provides contextual information to the accelerator, 
+while conditioning on prior sampled tokens allows it to produce higher-quality draft n-grams.
 
 ## Code