Fix MODEL_ENTRIES import

This commit is contained in:
Jash Gulabrai 2025-04-15 13:03:36 -04:00
parent 5f2f838656
commit 14420e880a

View file

@ -7,7 +7,6 @@ from typing import Any, Dict, List
import requests
from build.lib.llama_stack.providers.remote.inference.nvidia.models import MODEL_ENTRIES
from llama_stack.apis.agents import Agents
from llama_stack.apis.benchmarks import Benchmark
from llama_stack.apis.datasetio import DatasetIO
@ -15,6 +14,7 @@ from llama_stack.apis.datasets import Datasets
from llama_stack.apis.inference import Inference
from llama_stack.apis.scoring import Scoring, ScoringResult
from llama_stack.providers.datatypes import BenchmarksProtocolPrivate
from llama_stack.providers.remote.inference.nvidia.models import MODEL_ENTRIES
from llama_stack.providers.utils.inference.model_registry import ModelRegistryHelper
from .....apis.common.job_types import Job, JobStatus