From d9f75cc98fbb4172751c97e191ec8df819c92b2a Mon Sep 17 00:00:00 2001 From: Botao Chen Date: Thu, 2 Jan 2025 13:15:31 -0800 Subject: [PATCH] Import from the right path (#708) Import BaseModel and Field from pydantic --- llama_stack/apis/eval/eval.py | 3 ++- .../providers/inline/post_training/torchtune/common/utils.py | 4 ++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/llama_stack/apis/eval/eval.py b/llama_stack/apis/eval/eval.py index 2592bca37..1073d6310 100644 --- a/llama_stack/apis/eval/eval.py +++ b/llama_stack/apis/eval/eval.py @@ -6,9 +6,10 @@ from typing import Any, Dict, List, Literal, Optional, Protocol, Union -from llama_models.llama3.api.datatypes import BaseModel, Field from llama_models.schema_utils import json_schema_type, webmethod +from pydantic import BaseModel, Field + from typing_extensions import Annotated from llama_stack.apis.agents import AgentConfig 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 9673e0732..a5279cdbe 100644 --- a/llama_stack/providers/inline/post_training/torchtune/common/utils.py +++ b/llama_stack/providers/inline/post_training/torchtune/common/utils.py @@ -15,12 +15,12 @@ from typing import Any, Callable, Dict, List import torch from llama_models.datatypes import Model - -from llama_models.llama3.api.datatypes import BaseModel 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, lora_llama3_8b from torchtune.models.llama3._tokenizer import Llama3Tokenizer from torchtune.models.llama3_2 import lora_llama3_2_3b