diff --git a/examples/models/llama/UTILS.md b/examples/models/llama/UTILS.md index 27a7a5832d..dd014240ac 100644 --- a/examples/models/llama/UTILS.md +++ b/examples/models/llama/UTILS.md @@ -45,7 +45,7 @@ You can also download above models from [Hugging Face](https://huggingface.co/). ```Python -from torchtune.utils import FullModelHFCheckpointer +from torchtune.training.checkpointing import FullModelHFCheckpointer from torchtune.models import convert_weights import torch