RedHitMark commited on
Commit
40d53b6
·
verified ·
1 Parent(s): 3d97d52

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -64,8 +64,8 @@ Load SEMPL-IT model and tokenizer:
64
  ```py
65
  from transformers import AutoModelForCausalLM, AutoTokenizer
66
 
67
- tokenizer = AutoTokenizer.from_pretrained("VerbACxSS/sempl-it-proofreading")
68
- model = AutoModelForCausalLM.from_pretrained("VerbACxSS/sempl-it-proofreading").to("cuda")
69
  ```
70
 
71
  Define and apply chat template:
 
64
  ```py
65
  from transformers import AutoModelForCausalLM, AutoTokenizer
66
 
67
+ tokenizer = AutoTokenizer.from_pretrained("VerbACxSS/sempl-it-proofreading-bnb")
68
+ model = AutoModelForCausalLM.from_pretrained("VerbACxSS/sempl-it-proofreading-bnb").to("cuda")
69
  ```
70
 
71
  Define and apply chat template: