Skip to content

Commit c5d7ab3

Browse files
Update generate.py
1 parent 6b81264 commit c5d7ab3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

generate.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ def main(
2727
):
2828
assert (
2929
base_model
30-
), "Please specify a --base_model, e.g. --base_model='decapoda-research/llama-7b-hf'"
30+
), "Please specify a --base_model, e.g. --base_model='bigscience/bloomz-7b1-mt'"
3131

3232
tokenizer = BloomTokenizerFast.from_pretrained(base_model)
3333
if device == "cuda":

0 commit comments

Comments
 (0)