Update README.md
Browse files
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-
|
13 |
-
tokenizer = AutoTokenizer.from_pretrained("nolanoAI/lordcoder-v0-
|
14 |
|
15 |
inputs = {k: v.to(device) for k,v in tokenizer('# PyTorch CNN on MNIST\nimport torch\n', return_tensors='pt').items()}
|
16 |
|
@@ -22,7 +24,7 @@ generated_ids = model.generate(
|
|
22 |
top_p=0.95,
|
23 |
do_sample=True,
|
24 |
eos_token_id=tokenizer.eos_token_id,
|
25 |
-
pad_token_id=tokenizer.eos_token_id,
|
26 |
)
|
27 |
|
28 |
|
|
|
2 |
license: bigcode-openrail-m
|
3 |
---
|
4 |
|
5 |
+
## LoRDCoder v0 13.8B
|
6 |
+
|
7 |
Usage:
|
8 |
|
9 |
```
|
|
|
11 |
|
12 |
device = "cuda"
|
13 |
|
14 |
+
model = AutoModelForCausalLM.from_pretrained("nolanoAI/lordcoder-v0-13-8B", trust_remote_code=True).to(device)
|
15 |
+
tokenizer = AutoTokenizer.from_pretrained("nolanoAI/lordcoder-v0-13-8B", 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 |
|
|
|
24 |
top_p=0.95,
|
25 |
do_sample=True,
|
26 |
eos_token_id=tokenizer.eos_token_id,
|
27 |
+
pad_token_id=tokenizer.eos_token_id,
|
28 |
)
|
29 |
|
30 |
|