starmpcc commited on
Commit
32044a5
·
1 Parent(s): 579271b

Update Scrpit on ReadME

Browse files
Files changed (1) hide show
  1. README.md +3 -4
README.md CHANGED
@@ -83,10 +83,9 @@ The response should provide the accurate answer to the instruction, while being
83
  {question}
84
  [Instruction End]
85
  """
86
-
87
- from transformers import AutoTokenizer, AutoModel
88
- tokenizer = AutoTokenizer.from_pretrained("starmpcc/Asclepius-Llama2-13B")
89
- model = AutoModel.from_pretrained("starmpcc/Asclepius-Llama2-13B")
90
 
91
  note = "This is a sample note"
92
  question = "What is the diagnosis?"
 
83
  {question}
84
  [Instruction End]
85
  """
86
+ from transformers import AutoTokenizer, AutoModelForCausalLM
87
+ tokenizer = AutoTokenizer.from_pretrained("starmpcc/Asclepius-Llama2-13B", use_fast=False)
88
+ model = AutoModelForCausalLM.from_pretrained("starmpcc/Asclepius-Llama13-7B")
 
89
 
90
  note = "This is a sample note"
91
  question = "What is the diagnosis?"