Ayushk44 commited on
Commit
11432fd
1 Parent(s): 46d90bf

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -3
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
 
@@ -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, # model has no pad token
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