Ayushk44 commited on
Commit
c8f2357
1 Parent(s): f698051

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -2
README.md CHANGED
@@ -2,6 +2,8 @@
2
  license: bigcode-openrail-m
3
  ---
4
 
 
 
5
  Usage:
6
 
7
  ```
@@ -9,8 +11,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
9
 
10
  device = "cuda"
11
 
12
- model = AutoModelForCausalLM.from_pretrained("nolanoAI/lordcoder-v0-14-9B", trust_remote_code=True).to(device)
13
- tokenizer = AutoTokenizer.from_pretrained("nolanoAI/lordcoder-v0-14-9B", trust_remote_code=True)
14
 
15
  inputs = {k: v.to(device) for k,v in tokenizer('# PyTorch CNN on MNIST\nimport torch\n', return_tensors='pt').items()}
16
 
 
2
  license: bigcode-openrail-m
3
  ---
4
 
5
+ ## LoRDCoder v0 13.2B
6
+
7
  Usage:
8
 
9
  ```
 
11
 
12
  device = "cuda"
13
 
14
+ model = AutoModelForCausalLM.from_pretrained("nolanoAI/lordcoder-v0-13-2B", trust_remote_code=True).to(device)
15
+ tokenizer = AutoTokenizer.from_pretrained("nolanoAI/lordcoder-v0-13-2B", trust_remote_code=True)
16
 
17
  inputs = {k: v.to(device) for k,v in tokenizer('# PyTorch CNN on MNIST\nimport torch\n', return_tensors='pt').items()}
18