Edit model card

Model Card for Model ID

This model is a fine-tuned version of meta-llama/Llama-2-7b-hf on timdettmers/openassistant-guanaco dataset.

Model Details

Model Description

This is a fine-tuned version of the meta-llama/Llama-2-7b-hf model using Parameter Efficient Fine Tuning (PEFT) with Low Rank Adaptation (LoRA) on the Intel Gaudi 2 AI accelerator. This model can be used for various text generation tasks including chatbots, content creation, and other NLP applications.

  • Developed by: Migara Amarasinghe
  • Model type: LLM
  • Language(s) (NLP): English
  • Finetuned from model [optional]: meta-llama/Llama-2-7b-hf

Uses

Direct Use

This model can be used for text generation tasks such as:

  • Chatbots
  • Automated content creation
  • Text completion and augmentation

Out-of-Scope Use

  • Use in real-time applications where latency is critical
  • Use in highly sensitive domains without thorough evaluation and testing

Recommendations

Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.

Training Details

Training Hyperparameters

  • Training regime: Mixed precision training using bf16
  • Number of epochs: 3
  • Learning rate: 1e-4
  • Batch size: 16
  • Seq length: 512

Environmental Impact

Carbon emissions can be estimated using the Machine Learning Impact calculator presented in Lacoste et al. (2019).

  • Hardware Type: Intel Gaudi AI Accelerator
  • Hours used: < 1 hour

Technical Specifications

Compute Infrastructure

Hardware

  • Intel Gaudi 2 AI Accelerator
  • Intel(R) Xeon(R) Platinum 8368 CPU

Software

  • Transformers library
  • Optimum Habana library
Downloads last month

-

Downloads are not tracked for this model. How to track
Unable to determine this model’s pipeline type. Check the docs .

Dataset used to train migaraa/Gaudi_LoRA_Llama-2-7b-hf