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