Update README.md

README.md

@@ -21,15 +21,15 @@ import transformers
 model = transformers.AutoModelForCausalLM.from_pretrained(
     "miguelcarv/phi-2-slimorca",
     trust_remote_code=True,
-    torch_dtype = torch.bfloat16
+    torch_dtype=torch.bfloat16
 ).to('cuda')
 tokenizer = transformers.AutoTokenizer.from_pretrained("microsoft/phi-2")
 
 
-SYSTEM_PROMPT = "You are an AI assistant. You will be given a task. You must generate a detailed and long answer."
+SYSTEM_PROMPT = "You are an AI assistant. You will be given a task. You must generate a short and concise answer."
 input_text = f"""{SYSTEM_PROMPT}
 
-Instruction: Give me the first 5 prime numbers and explain what prime numbers are.
+Instruction: Give me the first 5 prime numbers and explain what prime numbers are. 
 Output:"""
 
 with torch.no_grad():
@@ -39,7 +39,7 @@ with torch.no_grad():
         num_beams = 3,
         eos_token_id = tokenizer.eos_token_id
     )
-print(tokenizer.decode(outputs[0], skip_special_tokens=False))
+print(tokenizer.decode(outputs[0], skip_special_tokens=True))
 ```
 
 ## Training Details