Compatibility with Llama 3.1 8B Models

Hello,

I've been using your implementation from the repository linked with the paper "How do Large Language Models Handle Multilingualism" and I found it very insightful and well-structured. Currently, the implementation is tailored for Llama 3 8B models. I am considering using Llama 3.1 8B models for my own research and development work.

Could you please let me know if there are any plans to update the repository to support Llama 3.1 8B models? This update would be extremely beneficial for those of us looking to leverage the latest advancements in Llama models.

Thank you for your great work and looking forward to your response!