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

datasets api

Browse source
This commit is contained in:
Xi Yan 2024-10-14 13:16:39 -07:00
parent 18fe966e96
commit f046899a1c
15 changed files with 281 additions and 80 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/distribution/registry/datasets/__init__.py
View file

@ -5,9 +5,9 @@
# the root directory of this source tree.
# TODO: make these import config based
from llama_stack.apis.dataset import * # noqa: F403
from llama_stack.apis.datasets import * # noqa: F403
from ..registry import Registry
from .dataset import CustomDataset, HuggingfaceDataset
from .dataset_wrappers import CustomDataset, HuggingfaceDataset
class DatasetRegistry(Registry[BaseDataset]):