Skip to content

Commit 1782c2d

Browse files
committed
correct tokenizer
1 parent 62fcebb commit 1782c2d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

evaluation/generation/generate.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ def main():
3434
args = get_args()
3535
print(f"Loading model", flush=True)
3636

37-
tokenizer = AutoTokenizer.from_pretrained("bigscience/bloom", padding_side="left")
37+
tokenizer = AutoTokenizer.from_pretrained(args.checkpoint, padding_side="left")
3838

3939
print("Loaded tokenizer !")
4040
start = datetime.datetime.now()

0 commit comments

Comments
 (0)