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

How to convert lit-llama pretrained model to HF format? #486

Open
karkeranikitha opened this issue Apr 16, 2024 · 1 comment
Open

How to convert lit-llama pretrained model to HF format? #486

karkeranikitha opened this issue Apr 16, 2024 · 1 comment

Comments

@karkeranikitha
Copy link

Hi

I have pretrained a model and have it in lit-llama format. How can I convert it to huggingface format? I need to load my pretrained model via HuggingFace for inference.
There are no scripts for lit-llama to HF format. Kindly suggest any approach or provide the script for the same.

Thanks in advance.

@karkeranikitha karkeranikitha changed the title How to convert litllama pretrained model to HF format? How to convert lit-llama pretrained model to HF format? Apr 16, 2024
@rasbt
Copy link
Contributor

rasbt commented Apr 16, 2024

I am not entirely sure, but https://github.com/Lightning-AI/lit-llama/blob/main/scripts/convert_checkpoint.py might be doing that

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

No branches or pull requests

2 participants