Update README.md

README.md

@@ -15,6 +15,8 @@ language:
 
 <!-- Provide a quick summary of what the model is/does. -->
 
+**For transformers versions v4.40.0 or newer, we suggest using [OLMo 7B SFT HF](https://huggingface.co/allenai/OLMo-7B-SFT-hf) instead.**
+
 OLMo is a series of **O**pen **L**anguage **Mo**dels designed to enable the science of language models.
 The OLMo base models are trained on the [Dolma](https://huggingface.co/datasets/allenai/dolma) dataset.
 The adapted versions are trained on the [Tulu SFT mixture](https://huggingface.co/datasets/allenai/tulu-v2-sft-mixture) and, for the Instruct version, a [cleaned version of the UltraFeedback dataset](https://huggingface.co/datasets/allenai/ultrafeedback_binarized_cleaned).
@@ -81,11 +83,9 @@ pip install ai2-olmo
 ```
 Now, proceed as usual with HuggingFace:
 ```python
-import hf_olmo
-
-from transformers import AutoModelForCausalLM, AutoTokenizer
-olmo = AutoModelForCausalLM.from_pretrained("allenai/OLMo-7B-SFT")
-tokenizer = AutoTokenizer.from_pretrained("allenai/OLMo-7B-SFT")
+from hf_olmo import OLMoForCausalLM, OLMoTokenizerFast
+olmo = OLMoForCausalLM.from_pretrained("allenai/OLMo-7B-SFT")
+tokenizer = OLMoTokenizerFast.from_pretrained("allenai/OLMo-7B-SFT")
 chat = [
     { "role": "user", "content": "What is language modeling?" },
 ]