jiangchengchengNLP commited on
Commit
b43b040
·
verified ·
1 Parent(s): 0b41faa

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -1
README.md CHANGED
@@ -42,7 +42,9 @@ OPT-6.7B
42
  from transformers import AutoModelForCausalLM, AutoTokenizer
43
 
44
  model_id = "jiangchengchengNLP/opt-6.7B-AWQ-8INT"
45
- tokenizer = AutoTokenizer.from_pretrained(model_id)
 
 
46
  model = AutoModelForCausalLM.from_pretrained(model_id)
47
 
48
  input_text = "Hello, I'm am conscious and"
 
42
  from transformers import AutoModelForCausalLM, AutoTokenizer
43
 
44
  model_id = "jiangchengchengNLP/opt-6.7B-AWQ-8INT"
45
+ tokenizer_id="facebook/opt-6.7b"
46
+
47
+ tokenizer = AutoTokenizer.from_pretrained(tokenizer_id)
48
  model = AutoModelForCausalLM.from_pretrained(model_id)
49
 
50
  input_text = "Hello, I'm am conscious and"