Update README.md

README.md

@@ -2,6 +2,8 @@
 license: bigcode-openrail-m
 ---
 
+## LoRDCoder v0 14.5B
+
 Usage:
 
 ```
@@ -9,8 +11,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 
 device = "cuda"
 
-model = AutoModelForCausalLM.from_pretrained("nolanoAI/lordcoder-v0-14-9B", trust_remote_code=True).to(device)
-tokenizer = AutoTokenizer.from_pretrained("nolanoAI/lordcoder-v0-14-9B", trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("nolanoAI/lordcoder-v0-14-5B", trust_remote_code=True).to(device)
+tokenizer = AutoTokenizer.from_pretrained("nolanoAI/lordcoder-v0-14-5B", trust_remote_code=True)
 
 inputs = {k: v.to(device) for k,v in tokenizer('# PyTorch CNN on MNIST\nimport torch\n', return_tensors='pt').items()}
 
@@ -22,7 +24,7 @@ generated_ids = model.generate(
         top_p=0.95,
         do_sample=True,
         eos_token_id=tokenizer.eos_token_id,
-        pad_token_id=tokenizer.eos_token_id,  # model has no pad token
+        pad_token_id=tokenizer.eos_token_id,
     )