RedHitMark commited on
Commit
012f4e9
·
verified ·
1 Parent(s): 27e0cc2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -79,8 +79,8 @@ Load SEMPL-IT model and tokenizer:
79
  ```py
80
  from transformers import AutoModelForCausalLM, AutoTokenizer
81
 
82
- tokenizer = AutoTokenizer.from_pretrained("VerbACxSS/sempl-it-lex-awq")
83
- model = AutoModelForCausalLM.from_pretrained("VerbACxSS/sempl-it-lex-awq").to("cuda")
84
  ```
85
 
86
  Define and apply chat template:
 
79
  ```py
80
  from transformers import AutoModelForCausalLM, AutoTokenizer
81
 
82
+ tokenizer = AutoTokenizer.from_pretrained("VerbACxSS/sempl-it-lex-bnb")
83
+ model = AutoModelForCausalLM.from_pretrained("VerbACxSS/sempl-it-lex-bnb").to("cuda")
84
  ```
85
 
86
  Define and apply chat template: