Skip to content

Commit c5a37bc

Browse files
authored
Rename 176B presets (#1496)
1 parent fe5a53b commit c5a37bc

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tools/checkpoint_conversion/convert_bloom_checkpoints.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -34,25 +34,25 @@
3434
"bloom_1.7b_multi": "bigscience/bloom-1b7",
3535
"bloom_3b_multi": "bigscience/bloom-3b",
3636
"bloom_7b_multi": "bigscience/bloom-7b1",
37-
"bloom_multi": "bigscience/bloom",
37+
"bloom_176b_multi": "bigscience/bloom",
3838
# Multitask finetuned on xP3 (Crosslingual Public Pool of Prompts) https://huggingface.co/datasets/bigscience/xP3
3939
# xP3 is a mixture of 13 training tasks in 46 languages with English prompts
4040
"bloomz_560m_multi": "bigscience/bloomz-560m",
4141
"bloomz_1.1b_multi": "bigscience/bloomz-1b1",
4242
"bloomz_1.7b_multi": "bigscience/bloomz-1b7",
4343
"bloomz_3b_multi": "bigscience/bloomz-3b",
4444
"bloomz_7b_multi": "bigscience/bloomz-7b1",
45-
"bloomz_multi": "bigscience/bloomz",
45+
"bloomz_176b_multi": "bigscience/bloomz",
4646
# Multitask finetuned on xP3mt
4747
# (Crosslingual Public Pool of Prompts machine-translated) https://huggingface.co/datasets/bigscience/xP3
4848
# xP3mt is Mixture of 13 training tasks in 46 languages with prompts in 20
4949
# languages (machine-translated from English)
5050
"bloomz_7b_mt": "bigscience/bloomz-7b1-mt",
51-
"bloomz_mt": "bigscience/bloomz-mt",
51+
"bloomz_176b_mt": "bigscience/bloomz-mt",
5252
# Multitask finetuned on P3 (Public Pool of Prompts) https://huggingface.co/datasets/Muennighoff/P3
5353
# xP3 is a mixture of 8 training tasks with English-only prompts
5454
"bloomz_7b_p3": "bigscience/bloomz-7b1-p3",
55-
"bloomz_p3": "bigscience/bloomz-p3",
55+
"bloomz_176b_p3": "bigscience/bloomz-p3",
5656
}
5757

5858
EXTRACT_DIR = "./model"

0 commit comments

Comments
 (0)