Update README.md

README.md

@@ -9,9 +9,9 @@ base_model:
 - microsoft/wavecoder-ultra-6.7b
 ---
 
-# NeuralPipe-7B-sl
+# CodeMaster v1 7b
 
-NeuralPipe-7B-sl is a merge of the following models using [LazyMergekit](https://colab.research.google.com/drive/1obulZ1ROXHjYLn6PPZJwRR6GzgQogxxb?usp=sharing):
+CodeMaster v1 7b is a merge of the following models using [LazyMergekit](https://colab.research.google.com/drive/1obulZ1ROXHjYLn6PPZJwRR6GzgQogxxb?usp=sharing):
 * [microsoft/wavecoder-ultra-6.7b](https://huggingface.co/microsoft/wavecoder-ultra-6.7b)
 * [microsoft/wavecoder-ultra-6.7b](https://huggingface.co/microsoft/wavecoder-ultra-6.7b)
 
@@ -59,7 +59,7 @@ from transformers import AutoTokenizer
 import transformers
 import torch
 
-model = "KingNish/NeuralPipe-7B-sl"
+model = "KingNish/CodeMaster-v1-7b"
 messages = [{"role": "user", "content": "What is a large language model?"}]
 
 tokenizer = AutoTokenizer.from_pretrained(model)
@@ -71,6 +71,6 @@ pipeline = transformers.pipeline(
     device_map="auto",
 )
 
-outputs = pipeline(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
+outputs = pipeline(prompt, max_new_tokens=8096, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
 print(outputs[0]["generated_text"])
 ```