Model Description

These are the peft layers of the finetuned codellama 7b model that is to be used as the retriever component of RAG (Retrieval Augmented Generation) for codebases

• Developed by: arcee.ai
• Model type: Retriever component for RAG for codebases
• Language(s) (NLP): English
• License: apache-2.0
• Finetuned from model: codellama/CodeLlama-7b-Instruct-hf

Training Details

• Trained on a small dataset of size: 160,000 query-passage pairs

• The pretrained model codellama/CodeLlama-7b-Instruct-hf was finetuned using data containing code

• PEFT was used but not quantization

• the hidden state associated with the EOS token was used as the embeddings

• Model was trained for 1 epoch

• PEFT 0.5.0