Update README.md

README.md

@@ -100,8 +100,8 @@ def generate_text(model, tokenizer, prompt, max_length=1024):
     output = model.generate(input_ids, 
                             max_length=max_length, 
                             do_sample=True, 
-                            top_k=50, 
-                            top_p=0.30,
+                            top_k=35, 
+                            top_p=0.80,
                             pad_token_id=tokenizer.pad_token_id,
                             eos_token_id=tokenizer.eos_token_id,
                             attention_mask=attention_mask)
@@ -119,4 +119,4 @@ while True:
     print(output_text)
 ```
 ## Deploying and training the model
-The model has been fine-tuned on a specific input format that goes like this ```"<|SYSTEM|> {system prompt} <|USER|> {user prompt} <|ASSISTANT|> {model prediction} <|End|>".``` For the best performance from the model the input text should be as follows ```<|SYSTEM|> {system prompt} <|USER|> {user prompt} <|ASSISTANT|> ``` and the target/label should be as follows ```<|SYSTEM|> {system prompt} <|USER|> {user prompt} <|ASSISTANT|> {dataset output} <|End|>```
+The model has been fine-tuned on a specific input format that goes like this ```"<|USER|> {user prompt} <|ASSISTANT|> {model prediction} <|End|>".``` For the best performance from the model the input text should be as follows ```<|USER|> {user prompt} <|ASSISTANT|> ``` and the target/label should be as follows ```<|USER|> {user prompt} <|ASSISTANT|> {dataset output} <|End|>```