phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-28 19:04:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Import from the right path (#708)

Browse source 
Import BaseModel and Field from pydantic
This commit is contained in:
Botao Chen 2025-01-02 13:15:31 -08:00 • committed by GitHub
parent 750604c7af
commit d9f75cc98f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

3
llama_stack/apis/eval/eval.py
View file

@ -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

4
llama_stack/providers/inline/post_training/torchtune/common/utils.py
View file

@ -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