Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add LoRA fine-tuning for PET #379

Merged
merged 39 commits into from
Nov 8, 2024
Merged

Add LoRA fine-tuning for PET #379

merged 39 commits into from
Nov 8, 2024

Conversation

abmazitov
Copy link
Contributor

@abmazitov abmazitov commented Nov 5, 2024

This PR adds the LoRA fine-tuning functionality to PET model. Also closes the issues: #375 and #360

Contributor (creator of pull-request) checklist

  • Tests updated (for new features and bugfixes)?
  • Documentation updated (for new features)?
  • Issue referenced (for PRs that solve an issue)?

📚 Documentation preview 📚: https://metatrain--379.org.readthedocs.build/en/379/

@abmazitov abmazitov marked this pull request as ready for review November 7, 2024 16:01
Comment on lines +132 to +133
else:
new_continue_from = continue_from
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks and sorry for the bug!

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

No worries!

What is LoRA?
^^^^^^^^^^^^^

LoRA (Low-Rank Adaptation) is a technique used to adapt pre-trained models
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
LoRA (Low-Rank Adaptation) is a technique used to adapt pre-trained models
LoRA (Low-Rank Adaptation) Parameter Efficient Fine Tuning (PEFT) is a technique
used to adapt pre-trained models

I would put also the definition of PEFT only becouse then the name appears in the parameters file, and an user that know nothig can be confused if it is a typo with PET.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Yes I agree, thank you!

@abmazitov abmazitov merged commit a69d9ab into main Nov 8, 2024
12 checks passed
@abmazitov abmazitov deleted the pet-fine-tuning branch November 8, 2024 15:15
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants