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

Laplace lora #11

Merged
merged 12 commits into from
Feb 14, 2024
Merged

Laplace lora #11

merged 12 commits into from
Feb 14, 2024

Conversation

phoebeklett
Copy link
Contributor

This PR will add an example of using Laplace LoRA with UQlib.

Namely, we'll compute post-hoc Laplace approximations to LoRA parameters at model checkpoints theta_MAP obtained from standard fine-tuning (using PEFT).

Model: Llama-2-7b
Data: https://huggingface.co/datasets/timdettmers/openassistant-guanaco

@farice
Copy link

farice commented Feb 6, 2024

Could this really be it...

@dpsimpson
Copy link

dpsimpson commented Feb 6, 2024 via email

@phoebeklett
Copy link
Contributor Author

@SamDuffield Training loop complete. Took out N (size of dataset) for now, until I can figure out a graceful way to incorporate.

@phoebeklett phoebeklett marked this pull request as ready for review February 13, 2024 17:30
Copy link
Contributor

@johnathanchiu johnathanchiu left a comment

Choose a reason for hiding this comment

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

Left some comments. Code looks fine overall.

experiments/laplace_lora/lora_transformer.py Outdated Show resolved Hide resolved
experiments/laplace_lora/lora_transformer.py Show resolved Hide resolved
experiments/laplace_lora/lora_transformer.py Outdated Show resolved Hide resolved
@phoebeklett phoebeklett merged commit 5de8c2f into main Feb 14, 2024
2 checks passed
@phoebeklett phoebeklett deleted the laplace-lora branch February 14, 2024 15:25
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.

5 participants