connectthapa84 commited on
Commit
05b3317
·
verified ·
1 Parent(s): eeb00bb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -33,8 +33,8 @@ BioMed-R1 can be used just like `Llama-3.1-8B-Instruct`. You can deploy it with
33
  ```python
34
  from transformers import AutoModelForCausalLM, AutoTokenizer
35
 
36
- model = AutoModelForCausalLM.from_pretrained("BioMed-R1",torch_dtype="auto",device_map="auto")
37
- tokenizer = AutoTokenizer.from_pretrained("BioMed-R1")
38
 
39
  input_text = "Does vagus nerve contribute to the development of steatohepatitis and obesity in phosphatidylethanolamine N-methyltransferase deficient mice?"
40
  messages = [{"role": "user", "content": input_text}]
 
33
  ```python
34
  from transformers import AutoModelForCausalLM, AutoTokenizer
35
 
36
+ model = AutoModelForCausalLM.from_pretrained("zou-lab/BioMed-R1-8B",torch_dtype="auto",device_map="auto")
37
+ tokenizer = AutoTokenizer.from_pretrained("zou-lab/BioMed-R1-8B")
38
 
39
  input_text = "Does vagus nerve contribute to the development of steatohepatitis and obesity in phosphatidylethanolamine N-methyltransferase deficient mice?"
40
  messages = [{"role": "user", "content": input_text}]