alfraser commited on
Commit
946c170
·
1 Parent(s): a9d1d49

Added a config to test V6 fine-tuned model

Browse files
Files changed (1) hide show
  1. config/architectures.json +9 -0
config/architectures.json CHANGED
@@ -30,6 +30,15 @@
30
  {"class": "OutputResponseScreener"}
31
  ],
32
  "img": "architecture_rag.jpg"
 
 
 
 
 
 
 
 
 
33
  }
34
  ]
35
  }
 
30
  {"class": "OutputResponseScreener"}
31
  ],
32
  "img": "architecture_rag.jpg"
33
+ },
34
+ {
35
+ "name": "4. Fine-tuning Architecture V6",
36
+ "description": "Model fine-tuned on the data from the baseline product dataset.",
37
+ "steps": [
38
+ {"class": "InputRequestScreener"},
39
+ {"class": "HFInferenceEndpoint", "params": {"endpoint_url": "https://ln8i9z4ecjqora6d.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"}},
40
+ {"class": "OutputResponseScreener"}
41
+ ]
42
  }
43
  ]
44
  }