lightning llama
sri9s opened this issue · comments
Are there lightning data and trainer modules available for llama?
trainer version is available here - https://github.com/Lightning-AI/lit-gpt/blob/main/pretrain/openwebtext_trainer.py
Implementation of the LLaMA language model based on nanoGPT. Supports flash attention, Int8 and GPTQ 4bit quantization, LoRA and LLaMA-Adapter fine-tuning, pre-training. Apache 2.0-licensed.
sri9s opened this issue · comments
Are there lightning data and trainer modules available for llama?
trainer version is available here - https://github.com/Lightning-AI/lit-gpt/blob/main/pretrain/openwebtext_trainer.py