gemma-2b-zephyr-dpo / README.md
tcapelle's picture
Update README.md
7e2818e verified
metadata
license: other
library_name: transformers
datasets:
  - HuggingFaceH4/ultrafeedback_binarized
base_model: wandb/gemma-2b-zephyr-sft
license_name: gemma-terms-of-use
license_link: https://ai.google.dev/gemma/terms

Visualize in Weights & Biases

Gemma 2B Zephyr DPO

The Zephyr DPO recipe applied on top of SFT finetuned Gemma 2B

Model description

  • Model type: A 8.5B parameter GPT-like model fine-tuned on a mix of publicly available, synthetic datasets.
  • Language(s) (NLP): Primarily English
  • Finetuned from model: wandb/gemma-2b-zephyr-sft

Recipe

We trained using the DPO script in alignment handbook recipe and logging to W&B

Visit the W&B workspace here

License

This model has the same license as the original Gemma model collection

Compute provided by Lambda Labs - 8xA100 80GB node

around 13 hours of training