From c8e14758ce76ba2e0928868ca17582135ab6954b Mon Sep 17 00:00:00 2001 From: 51616 Date: Tue, 10 Jun 2025 15:40:30 +0000 Subject: [PATCH] fix load_state_dict --- src/ctx_to_lora/modeling/hypernet.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/ctx_to_lora/modeling/hypernet.py b/src/ctx_to_lora/modeling/hypernet.py index 5187694..62c6539 100644 --- a/src/ctx_to_lora/modeling/hypernet.py +++ b/src/ctx_to_lora/modeling/hypernet.py @@ -673,7 +673,8 @@ class ModulatedPretrainedModel(nn.Module): return sd load_result = self.hypernet.load_state_dict( - remove_compile_prefix(state_dict), strict=True, *args, **kwargs + remove_compile_prefix(state_dict), + strict=True, # , *args, **kwargs ) logger.info(f"load result: {load_result}") return load_result