From 3105fac6dc25e6b7590e674aae3d98288c5f4818 Mon Sep 17 00:00:00 2001
From: 51616 <rujikorn.ch@gmail.com>
Date: Mon, 15 Sep 2025 01:37:28 +0900
Subject: [PATCH] remove num_gen_q

---
 run_eval.py                   | 6 ------
 src/ctx_to_lora/eval_utils.py | 3 ---
 2 files changed, 9 deletions(-)

diff --git a/run_eval.py b/run_eval.py
index a10a3f0..fa7977d 100644
--- a/run_eval.py
+++ b/run_eval.py
@@ -102,12 +102,6 @@ if __name__ == "__main__":
         action="store_true",
         help="Use generated queries for context distillation training.",
     )
-    parser.add_argument(
-        "--num_gen_q",
-        type=int,
-        default=None,
-        help="Number of generated queries to use for context distillation.",
-    )
     parser.add_argument(
         "--use_iterative_mode",
         action="store_true",
diff --git a/src/ctx_to_lora/eval_utils.py b/src/ctx_to_lora/eval_utils.py
index 8ddeb56..b484ed9 100644
--- a/src/ctx_to_lora/eval_utils.py
+++ b/src/ctx_to_lora/eval_utils.py
@@ -828,7 +828,6 @@ def evaluate(
                 )
                 ctx_distill_kwargs["q_model"] = q_model
                 ctx_distill_kwargs["q_tokenizer"] = q_tokenizer
-                ctx_distill_kwargs["num_gen_q"] = args.num_gen_q
             model = CtxDistillModel(peft_model, **ctx_distill_kwargs)
 
             add_tracker(model._distill_context, "distill_context")
@@ -1036,7 +1035,6 @@ def run_eval(
     use_cd: bool = False,
     cd_update_iterations: int = 10,
     cd_use_gen_q: bool = False,
-    num_gen_q: int = 20,
     use_iterative_mode: bool = False,
     use_llmlingua: bool = False,
     llmlingua_compression_rate: float = 0.9,
@@ -1098,7 +1096,6 @@ def run_eval(
             args.use_cd = use_cd
             args.cd_update_iterations = cd_update_iterations
             args.cd_use_gen_q = cd_use_gen_q
-            args.num_gen_q = num_gen_q
         if use_llmlingua:
             args.use_llmlingua = use_llmlingua
             args.llmlingua_compression_rate = llmlingua_compression_rate