YaoLuzjut commited on
Commit
a850442
·
verified ·
1 Parent(s): 71f1637

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -18,7 +18,7 @@ import torch
18
  from transformers import AutoTokenizer, AutoModelForCausalLM
19
 
20
  # Load the tokenizer and model
21
- model_path = "YaoLuzjut/Llama-3.1-6.3B-It-Dolly"
22
  tokenizer = AutoTokenizer.from_pretrained(model_path)
23
 
24
  device = 'cuda'
 
18
  from transformers import AutoTokenizer, AutoModelForCausalLM
19
 
20
  # Load the tokenizer and model
21
+ model_path = "YaoLuzjut/Llama-3.1-7.2B-It-Dolly"
22
  tokenizer = AutoTokenizer.from_pretrained(model_path)
23
 
24
  device = 'cuda'