Spaces:
Runtime error
Runtime error
Added the safety components to the fine-tuning model to make the test fair
Browse files
config/architectures.json
CHANGED
@@ -14,7 +14,9 @@
|
|
14 |
"name": "2. Fine-tuning Architecture",
|
15 |
"description": "Model fine-tuned on the data from the baseline product dataset.",
|
16 |
"steps": [
|
17 |
-
{"class": "
|
|
|
|
|
18 |
]
|
19 |
},
|
20 |
{
|
|
|
14 |
"name": "2. Fine-tuning Architecture",
|
15 |
"description": "Model fine-tuned on the data from the baseline product dataset.",
|
16 |
"steps": [
|
17 |
+
{"class": "InputRequestScreener"},
|
18 |
+
{"class": "HFInferenceEndpoint", "params": {"endpoint_url": "https://pgzu02dvzupp5sml.eu-west-1.aws.endpoints.huggingface.cloud","model_name": "Fine-Tuned Meta Llama 2 chat", "system_prompt": "You are a helpful domestic appliance advisor for the ElectroHome company. Please answer customer questions and do not mention other brands. If you cannot answer please say so.", "max_new_tokens": 1000, "prompt_style": "multi_line_with_roles"}},
|
19 |
+
{"class": "OutputResponseScreener"}
|
20 |
]
|
21 |
},
|
22 |
{
|