Skip to content

Commit 02239b3

Browse files
committed
Add comment as to why resetting model is needed
1 parent 54388dc commit 02239b3

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

src/heretic/main.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -738,6 +738,8 @@ def count_completed_trials() -> int:
738738
for name, value in get_trial_parameters(trial).items():
739739
print(f" * {name} = [bold]{value}[/]")
740740

741+
# Per https://github.com/huggingface/peft/issues/868#issuecomment-1820642893 once a LoRA is merged it's
742+
# expected to be empty. Provide a utility function to restore the previous LoRA-ified state.
741743
def reset_trial_model() -> None:
742744
print("* Resetting model...")
743745
model.reset_model()

0 commit comments

Comments
 (0)