Views
No views yet

1slices:
2- sources:
3 - layer_range: [0, 42]
4 model: meta-llama/Meta-Llama-3.1-405B-Instruct
5- sources:
6 - layer_range: [21, 63]
7 model: meta-llama/Meta-Llama-3.1-405B-Instruct
8- sources:
9 - layer_range: [42, 84]
10 model: meta-llama/Meta-Llama-3.1-405B-Instruct
11- sources:
12 - layer_range: [63, 105]
13 model: meta-llama/Meta-Llama-3.1-405B-Instruct
14- sources:
15 - layer_range: [84, 126]
16 model: meta-llama/Meta-Llama-3.1-405B-Instruct
17merge_method: passthrough
18dtype: bfloat161def generate_yaml_config(range_size, total_layers, nb_parameters):
2 new_size = total_layers + total_layers - range_size
3 new_param = (nb_parameters / total_layers) * new_size
4 print(f"New size = {new_size} layers")
5 print(f"New parameters = {new_param:.2f}B")
6 yaml_str = "slices:\n"
7
8 for i in range(0, round(total_layers - range_size + 1), range_size // 2):
9 start = i
10 end = min(start + range_size, total_layers)
11 yaml_str += f"- sources:\n"
12 yaml_str += f" - layer_range: [{start}, {end}]\n"
13 yaml_str += f" model: meta-llama/Meta-Llama-3.1-405B-Instruct\n"
14
15 yaml_str += "merge_method: passthrough\n"
16 yaml_str += "dtype: bfloat16\n"
17
18 print(yaml_str)
19
20 return new_size, new_param
21
22# Example usage
23new_size, new_param = generate_yaml_config(42, 126, 410)
24new_size, new_param = generate_yaml_config(105, new_size, new_param)