dnhkng commited on
Commit
e675f33
·
verified ·
1 Parent(s): eb613d0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -15,11 +15,11 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
15
  device = "cuda" # the device to load the model onto
16
 
17
  model = AutoModelForCausalLM.from_pretrained(
18
- "dnhkng/RYS-Huge",
19
  torch_dtype="auto",
20
  device_map="auto"
21
  )
22
- tokenizer = AutoTokenizer.from_pretrained("dnhkng/RYS-Huge")
23
 
24
  prompt = "Give me a short introduction to large language model."
25
  messages = [
 
15
  device = "cuda" # the device to load the model onto
16
 
17
  model = AutoModelForCausalLM.from_pretrained(
18
+ "dnhkng/RYS-XLarge",
19
  torch_dtype="auto",
20
  device_map="auto"
21
  )
22
+ tokenizer = AutoTokenizer.from_pretrained("dnhkng/RYS-XLarge")
23
 
24
  prompt = "Give me a short introduction to large language model."
25
  messages = [