qnguyen3 commited on
Commit
e754251
1 Parent(s): ab3d151

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -37,11 +37,11 @@ import torch
37
  device = "cuda" # the device to load the model onto
38
 
39
  model = AutoModelForCausalLM.from_pretrained(
40
- "vilm/VinaLlama2-14B",
41
  torch_dtype='auto',
42
  device_map="auto"
43
  )
44
- tokenizer = AutoTokenizer.from_pretrained("vilm/VinaLlama2-14B")
45
 
46
  prompt = "What is the mearning of life?"
47
  messages = [
 
37
  device = "cuda" # the device to load the model onto
38
 
39
  model = AutoModelForCausalLM.from_pretrained(
40
+ "qnguyen3/Master-Yi-9B",
41
  torch_dtype='auto',
42
  device_map="auto"
43
  )
44
+ tokenizer = AutoTokenizer.from_pretrained("qnguyen3/Master-Yi-9B")
45
 
46
  prompt = "What is the mearning of life?"
47
  messages = [