from local

This commit is contained in:
dohe0342 2023-05-24 21:19:54 +09:00
parent 8513a0d635
commit 7be2b54d72
2 changed files with 2 additions and 1 deletions

View File

@ -697,7 +697,8 @@ def main():
lora_modules.append(LoRAHook(module))
for i, lora in enumerate(lora_modules):
lora_param = torch.load(f"{params.exp_dir}/lora_{i}.pt")
lora.lora.load_state_dict(lora_param)
else:
if not params.use_averaged_model:
if params.iter > 0: