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

fix

Browse source
This commit is contained in:
Xi Yan 2024-11-07 18:25:39 -08:00
parent 33b6d9b7b7
commit 6ee02ca23b
6 changed files with 100 additions and 87 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/providers/inline/huggingface/datasetio/dataset_defs/llamastack_mmlu.py
View file

@ -5,7 +5,7 @@
# the root directory of this source tree.
from llama_models.llama3.api.datatypes import URL
from llama_stack.apis.common.type_system import StringType
from llama_stack.apis.common.type_system import CompletionInputType, StringType
from llama_stack.apis.datasetio import DatasetDef
@ -15,7 +15,7 @@ llamastack_mmlu = DatasetDef(
dataset_schema={
"expected_answer": StringType(),
"input_query": StringType(),
"generated_answer": StringType(),
"chat_completion_input": CompletionInputType(),
},
metadata={"path": "yanxi0830/ls-mmlu", "split": "train"},
)