alfraser commited on
Commit
ee60fb2
·
1 Parent(s): c6ae5fd

Added possibility to change the prompt style via parameter as I try and debug the fine-tuning issue.

Browse files
Files changed (1) hide show
  1. config/architectures.json +36 -0
config/architectures.json CHANGED
@@ -31,6 +31,42 @@
31
  {"class": "OutputResponseScreener"}
32
  ],
33
  "img": "architecture_rag.jpg"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
34
  }
35
  ]
36
  }
 
31
  {"class": "OutputResponseScreener"}
32
  ],
33
  "img": "architecture_rag.jpg"
34
+ },
35
+ {
36
+ "name": "4. Fine-tuning test 2 - multi_line prompt",
37
+ "description": "Fine-tuning to try different training methods",
38
+ "steps": [
39
+ {"class": "InputRequestScreener"},
40
+ {"class": "HFInferenceEndpoint", "params": {"endpoint_url": "https://d0on8skmq6t2a4wo.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"}},
41
+ {"class": "OutputResponseScreener"}
42
+ ]
43
+ },
44
+ {
45
+ "name": "5. Fine-tuning test 2 - multi_line_no_sys prompt",
46
+ "description": "Fine-tuning to try different training methods",
47
+ "steps": [
48
+ {"class": "InputRequestScreener"},
49
+ {"class": "HFInferenceEndpoint", "params": {"endpoint_url": "https://d0on8skmq6t2a4wo.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_no_sys"}},
50
+ {"class": "OutputResponseScreener"}
51
+ ]
52
+ },
53
+ {
54
+ "name": "6. Fine-tuning test 2 - single_line prompt",
55
+ "description": "Fine-tuning to try different training methods",
56
+ "steps": [
57
+ {"class": "InputRequestScreener"},
58
+ {"class": "HFInferenceEndpoint", "params": {"endpoint_url": "https://d0on8skmq6t2a4wo.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": "single_line"}},
59
+ {"class": "OutputResponseScreener"}
60
+ ]
61
+ },
62
+ {
63
+ "name": "7. Fine-tuning test 2 - single_line_no_sys prompt",
64
+ "description": "Fine-tuning to try different training methods",
65
+ "steps": [
66
+ {"class": "InputRequestScreener"},
67
+ {"class": "HFInferenceEndpoint", "params": {"endpoint_url": "https://d0on8skmq6t2a4wo.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": "single_line_no_sys"}},
68
+ {"class": "OutputResponseScreener"}
69
+ ]
70
  }
71
  ]
72
  }