Edit model card

My_AGI_llama_2_7B

Model Type: Fine-Tuned

Model Base: meta-llama/Llama-2-7b-hf

Datasets Used: databricks/databricks-dolly-15k

Author: Yuri Achermann

Date: June 03, 2024


Training procedure

Training Hyperparameters

The following hyperparameters were used during training:

  • learning_rate: 1e-05
  • train_batch_size: 2
  • eval_batch_size: 8
  • seed: 42
  • gradient_accumulation_steps: 4
  • total_train_batch_size: 8
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: linear
  • lr_scheduler_warmup_ratio: 0.05
  • training_steps: 593

Framework versions

  • PEFT==0.11.1
  • Transformers==4.41.2
  • Pytorch==2.1.0.post0+cxx11.abi
  • Datasets==2.19.2
  • Tokenizers==0.19.1

Intended uses & limitations

Primary Use Case: The model is intended for generating human-like responses in conversational applications, like chatbots or virtual assistants.

Limitations: The model may generate inaccurate or biased content as it reflects the data it was trained on. It is essential to evaluate the generated responses in context and use the model responsibly.


Evaluation

The evaluation platform consists of Gaudi Accelerators and Xeon CPUs running benchmarks from the Eleuther AI Language Model Evaluation Harness

Average ARC HellaSwag MMLU TruthfulQA Winogrande
54.904 45.65 76.8 42.02 40.2 69.85

Ethical Considerations

The model may inherit biases present in the training data. It is crucial to use the model in a way that promotes fairness and mitigates potential biases.


Acknowledgments

This fine-tuning effort was made possible by the support of Intel, that provided the computing resources, and Eduardo Alvarez. Additional shout-out to the creators of the Llama-2-7b-hf model and the contributors to the databricks-dolly-15k dataset.


Contact Information

For questions or feedback about this model, please contact Yuri Achermann.


License

This model is distributed under Apache 2.0 License.

Downloads last month
9
Inference Examples
Inference API (serverless) does not yet support peft models for this pipeline type.

Model tree for yuriachermann/My_AGI_llama_2_7B

Adapter
(1094)
this model

Dataset used to train yuriachermann/My_AGI_llama_2_7B

Space using yuriachermann/My_AGI_llama_2_7B 1

Collection including yuriachermann/My_AGI_llama_2_7B