phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-20 02:46:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

persist local hf datasetio

Browse source
This commit is contained in:
Xi Yan 2024-11-13 23:44:48 -05:00
parent e90ea1ab1e
commit 04ff184b34
2 changed files with 31 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

12
llama_stack/providers/adapters/datasetio/huggingface/config.py
View file

@ -3,7 +3,15 @@
#
# This source code is licensed under the terms described in the LICENSE file in
# the root directory of this source tree.
from llama_stack.apis.datasetio import * # noqa: F401, F403
from llama_stack.distribution.utils.config_dirs import RUNTIME_BASE_DIR
from llama_stack.providers.utils.kvstore.config import (
KVStoreConfig,
SqliteKVStoreConfig,
)
from pydantic import BaseModel
class HuggingfaceDatasetIOConfig(BaseModel): ...
class HuggingfaceDatasetIOConfig(BaseModel):
kvstore: KVStoreConfig = SqliteKVStoreConfig(
db_path=(RUNTIME_BASE_DIR / "huggingface_datasetio.db").as_posix()
) # Uses SQLite config specific to FAISS storage