From 316b404d948a492e9faebe77de1ebb55d120074f Mon Sep 17 00:00:00 2001
From: teleprint-me <77757836+teleprint-me@users.noreply.github.com>
Date: Sat, 18 May 2024 23:59:22 -0400
Subject: [PATCH] patch: Fix CLI option for generating vocab tests

---
 gguf-py/scripts/gguf-gen-pre.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/gguf-py/scripts/gguf-gen-pre.py b/gguf-py/scripts/gguf-gen-pre.py
index 7991d6344..dfdb4347b 100644
--- a/gguf-py/scripts/gguf-gen-pre.py
+++ b/gguf-py/scripts/gguf-gen-pre.py
@@ -177,7 +177,7 @@ def main():
     hf_vocab_req.generate_checksums()
     hf_vocab_req.log_pre_tokenizer_info()
 
-    if args.gen_tests:
+    if args.gen_vocab_tests:
         test_pre_tok(hf_vocab_req)
 
     if args.gen_vocab_script: