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

fix load_state_dict

Browse source
This commit is contained in:
51616 2025-06-10 15:40:30 +00:00
parent 4a2960cd64
commit c8e14758ce
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/ctx_to_lora/modeling/hypernet.py
View file

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