alfraser commited on
Commit
2cc68c2
·
1 Parent(s): d6b7bf0

Added the safety components to the fine-tuning model to make the test fair

Browse files
Files changed (1) hide show
  1. config/architectures.json +3 -1
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": "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"}}
 
 
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
  {