more fixes

This commit is contained in:
Xi Yan 2024-12-26 18:30:42 -08:00
parent 6596caed55
commit 27da763af9
10 changed files with 44 additions and 26 deletions

View file

@ -32,11 +32,11 @@ from llama_models.llama3.reference_impl.multimodal.model import (
CrossAttentionTransformer,
)
from llama_models.sku_list import resolve_model
from pydantic import BaseModel
from llama_stack.apis.inference import * # noqa: F403
from lmformatenforcer import JsonSchemaParser, TokenEnforcer, TokenEnforcerTokenizerData
from pydantic import BaseModel
from llama_stack.apis.inference import ResponseFormat, ResponseFormatType
from llama_stack.distribution.utils.model_utils import model_local_dir
from llama_stack.providers.utils.inference.prompt_adapter import (