apunkt/doc-to-lora
1
0
Fork 0
mirror of https://github.com/SakanaAI/doc-to-lora.git synced 2026-07-23 17:01:04 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

remove num_gen_q

Browse source
This commit is contained in:
51616 2025-09-15 01:37:28 +09:00
parent b77aa4a9f6
commit 3105fac6dc
2 changed files with 0 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

6
run_eval.py
View file

@ -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",

3
src/ctx_to_lora/eval_utils.py
View file

@ -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