Does EasyLM work for LLama-2?

Hello

Does this EasyLM also support fine-tuning LLama-2 7B on TPUs? I could only find support for Llama v1 but not v2. In particular, does the following script also work for v2? https://github.com/young-geng/EasyLM/blob/main/EasyLM/models/llama/convert_easylm_to_hf.py