Update README.md

README.md

@@ -43,8 +43,8 @@ Then:
 ```
 from open_lm.hf import *
 from transformers import AutoTokenizer, AutoModelForCausalLM
-tokenizer = AutoTokenizer.from_pretrained("apple/DCLM-Baseline-7B")
-model = AutoModelForCausalLM.from_pretrained("apple/DCLM-Baseline-7B")
+tokenizer = AutoTokenizer.from_pretrained("apple/DCLM-Baseline-7B-8k")
+model = AutoModelForCausalLM.from_pretrained("apple/DCLM-Baseline-7B-8k")
 
 inputs = tokenizer(["Machine learning is"], return_tensors="pt")
 gen_kwargs = {"max_new_tokens": 50, "top_p": 0.8, "temperature": 0.8, "do_sample": True, "repetition_penalty": 1.1}