mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-07-29 15:23:51 +00:00
excel dataset
This commit is contained in:
parent
5b7d24b1c3
commit
a56ea48d71
2 changed files with 16 additions and 3 deletions
|
@ -33,7 +33,12 @@ class CustomDataset(BaseDataset):
|
|||
def load(self):
|
||||
if self.dataset:
|
||||
return
|
||||
# TODO: better support w/ data url
|
||||
if self.url.endswith(".csv"):
|
||||
df = pandas.read_csv(self.url)
|
||||
elif self.url.endswith(".xlsx"):
|
||||
df = pandas.read_excel(self.url)
|
||||
|
||||
self.dataset = Dataset.from_pandas(df)
|
||||
|
||||
|
||||
|
|
|
@ -41,12 +41,20 @@ routing_table:
|
|||
inference:
|
||||
- provider_type: meta-reference
|
||||
config:
|
||||
model: Llama3.2-1B-Instruct
|
||||
model: Llama3.1-8B-Instruct
|
||||
quantization: null
|
||||
torch_seed: null
|
||||
max_seq_len: 4096
|
||||
max_batch_size: 1
|
||||
routing_key: Llama3.2-1B-Instruct
|
||||
routing_key: Llama3.1-8B-Instruct
|
||||
- provider_type: meta-reference
|
||||
config:
|
||||
model: Llama-Guard-3-1B
|
||||
quantization: null
|
||||
torch_seed: null
|
||||
max_seq_len: 4096
|
||||
max_batch_size: 1
|
||||
routing_key: Llama-Guard-3-1B
|
||||
# - provider_type: remote::tgi
|
||||
# config:
|
||||
# url: http://127.0.0.1:5009
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue