Update README.md

README.md

@@ -7,11 +7,13 @@ tags:
 base_model:
 - KingNish/CodeMaster-v1-7b
 - KingNish/CodeMaster-v1-7b
+license: mit
+pipeline_tag: text-generation
 ---
 
-# CodeMaster-v1-10b
+# CodeMaster-v1-9b
 
-CodeMaster-v1-10b is a merge of the following models using [LazyMergekit](https://colab.research.google.com/drive/1obulZ1ROXHjYLn6PPZJwRR6GzgQogxxb?usp=sharing):
+CodeMaster-v1-9b is a merge of the following models using [LazyMergekit](https://colab.research.google.com/drive/1obulZ1ROXHjYLn6PPZJwRR6GzgQogxxb?usp=sharing):
 * [KingNish/CodeMaster-v1-7b](https://huggingface.co/KingNish/CodeMaster-v1-7b)
 * [KingNish/CodeMaster-v1-7b](https://huggingface.co/KingNish/CodeMaster-v1-7b)
 
@@ -38,7 +40,7 @@ from transformers import AutoTokenizer
 import transformers
 import torch
 
-model = "KingNish/CodeMaster-v1-10b"
+model = "KingNish/CodeMaster-v1-9b"
 messages = [{"role": "user", "content": "What is a large language model?"}]
 
 tokenizer = AutoTokenizer.from_pretrained(model)
@@ -50,6 +52,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=8192, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
 print(outputs[0]["generated_text"])
 ```