Update README.md

README.md

@@ -64,7 +64,7 @@ Then you can load the model using HF's Auto classes as follows:
 from open_lm.hf import *
 from transformers import AutoTokenizer, AutoModelForCausalLM
 tokenizer = AutoTokenizer.from_pretrained("TRI-ML/DCLM-1B-IT")
-model = AutoModelForCausalLM.from_pretrained("TRI-ML/DCLM-1B")
+model = AutoModelForCausalLM.from_pretrained("TRI-ML/DCLM-1B-IT")
 
 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}