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

Use different positional embeddings #225

Open
aflah02 opened this issue Aug 28, 2024 · 2 comments
Open

Use different positional embeddings #225

aflah02 opened this issue Aug 28, 2024 · 2 comments

Comments

@aflah02
Copy link

aflah02 commented Aug 28, 2024

Hi
Really cool repo. Is there any way I can use different positional encodings during training? I want to essentially train several versions of tiny llama with say rope, absolute, and no positional encodings.

@zzhhjjj
Copy link
Collaborator

zzhhjjj commented Sep 3, 2024

I think this will require you to modify the code a little bit.

@aflah02
Copy link
Author

aflah02 commented Sep 4, 2024

I think this will require you to modify the code a little bit.

Can you point me to where you think these changes need to be done? Should I like a new copy of https://github.com/huggingface/nanotron/blob/main/src/nanotron/models/llama.py with different encodings? or should I be doing this elsewhere?

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