yue-maincode commited on
Commit
0a641f6
·
verified ·
1 Parent(s): 822adec

Update README.md (whole-number model naming)

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -75,7 +75,7 @@ parent; it is not a verified latency or cost measurement. Full tables and figure
75
  import torch
76
  from transformers import AutoModelForCausalLM, AutoTokenizer
77
 
78
- repo = "Maincode/matilda-platypus-1.6b-base"
79
  tok = AutoTokenizer.from_pretrained(repo)
80
  model = AutoModelForCausalLM.from_pretrained(repo, trust_remote_code=True, dtype=torch.bfloat16).cuda()
81
  inputs = tok("The capital of Australia is", return_tensors="pt").to("cuda")
 
75
  import torch
76
  from transformers import AutoModelForCausalLM, AutoTokenizer
77
 
78
+ repo = "Maincode/Matilda-Platypus-1.6B-Base"
79
  tok = AutoTokenizer.from_pretrained(repo)
80
  model = AutoModelForCausalLM.from_pretrained(repo, trust_remote_code=True, dtype=torch.bfloat16).cuda()
81
  inputs = tok("The capital of Australia is", return_tensors="pt").to("cuda")