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

added tiny llama examples for lora and qlora #1027

Merged
merged 2 commits into from
Jan 3, 2024

Conversation

tdolan21
Copy link
Contributor

@tdolan21 tdolan21 commented Jan 1, 2024

example for TinyLlama1.1B using lora and qlora. Completes in roughly 10 minutes on a 4090.

Comment on lines 66 to 69
special_tokens:
bos_token: "<s>"
eos_token: "</s>"
unk_token: "<unk>"
Copy link
Collaborator

Choose a reason for hiding this comment

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

Suggested change
special_tokens:
bos_token: "<s>"
eos_token: "</s>"
unk_token: "<unk>"
special_tokens:

these are un-necessary

Comment on lines 64 to 67
special_tokens:
bos_token: "<s>"
eos_token: "</s>"
unk_token: "<unk>"
Copy link
Collaborator

Choose a reason for hiding this comment

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

Suggested change
special_tokens:
bos_token: "<s>"
eos_token: "</s>"
unk_token: "<unk>"
special_tokens:

@winglian
Copy link
Collaborator

winglian commented Jan 2, 2024

can you also remove the existing tiny-llama yml here too pls? https://github.com/OpenAccess-AI-Collective/axolotl/blob/main/examples/llama-2/tiny-llama.yml

Copy link
Collaborator

@winglian winglian left a comment

Choose a reason for hiding this comment

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

Thank you!

@ehartford
Copy link
Collaborator

Awesome!

@winglian winglian merged commit c75f916 into axolotl-ai-cloud:main Jan 3, 2024
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.

None yet

3 participants