From da22796b49618ec14e64fd25fc4a333df76a8cba Mon Sep 17 00:00:00 2001 From: Botao Chen Date: Thu, 2 Jan 2025 11:48:45 -0800 Subject: [PATCH] temp commit --- .../providers/inline/post_training/torchtune/common/utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/llama_stack/providers/inline/post_training/torchtune/common/utils.py b/llama_stack/providers/inline/post_training/torchtune/common/utils.py index 6e48b78e4..8c73a1397 100644 --- a/llama_stack/providers/inline/post_training/torchtune/common/utils.py +++ b/llama_stack/providers/inline/post_training/torchtune/common/utils.py @@ -18,6 +18,7 @@ from llama_models.datatypes import Model from llama_models.sku_list import resolve_model from llama_stack.apis.common.type_system import ParamType, StringType from llama_stack.apis.datasets import Datasets +from pydantic import BaseModel from torchtune.models.llama3 import llama3_tokenizer from torchtune.models.llama3._tokenizer import Llama3Tokenizer