We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 6b81264 commit c5d7ab3Copy full SHA for c5d7ab3
generate.py
@@ -27,7 +27,7 @@ def main(
27
):
28
assert (
29
base_model
30
- ), "Please specify a --base_model, e.g. --base_model='decapoda-research/llama-7b-hf'"
+ ), "Please specify a --base_model, e.g. --base_model='bigscience/bloomz-7b1-mt'"
31
32
tokenizer = BloomTokenizerFast.from_pretrained(base_model)
33
if device == "cuda":
0 commit comments