From 64cc7d1b40d8c64198dc1c3b4a478753fad265d6 Mon Sep 17 00:00:00 2001 From: Botao Chen Date: Thu, 19 Dec 2024 23:50:49 -0800 Subject: [PATCH] temp commit --- llama_stack/providers/inline/eval/meta_reference/eval.py | 2 +- llama_stack/templates/experimental-post-training/run.yaml | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/llama_stack/providers/inline/eval/meta_reference/eval.py b/llama_stack/providers/inline/eval/meta_reference/eval.py index 453215e41..e1c2cc804 100644 --- a/llama_stack/providers/inline/eval/meta_reference/eval.py +++ b/llama_stack/providers/inline/eval/meta_reference/eval.py @@ -15,7 +15,7 @@ from llama_stack.apis.agents import Agents from llama_stack.apis.datasetio import DatasetIO from llama_stack.apis.datasets import Datasets from llama_stack.apis.eval_tasks import EvalTask -from llama_stack.apis.inference import Inference +from llama_stack.apis.inference import Inference, UserMessage from llama_stack.apis.scoring import Scoring from llama_stack.providers.datatypes import EvalTasksProtocolPrivate from llama_stack.providers.utils.kvstore import kvstore_impl diff --git a/llama_stack/templates/experimental-post-training/run.yaml b/llama_stack/templates/experimental-post-training/run.yaml index ad8551f91..3f390d83c 100644 --- a/llama_stack/templates/experimental-post-training/run.yaml +++ b/llama_stack/templates/experimental-post-training/run.yaml @@ -11,6 +11,7 @@ apis: - safety - scoring - telemetry +- post_training providers: inference: - provider_id: meta-reference-inference