Finetuning

#22
by omarabb315 - opened

Is there any tutorial or notebook to finetune the model?

Google org

Hi @omarabb315
Currently, the Gemma Cookbook contains three tutorials on fine-tuning the Gemma model:

Thanks this is great, can you please tell me the difference between those methods?

Google org
edited Aug 9

Basically, the notebooks utilize different datasets and toolkits—XTuner, Axolotl, and LLaMA Factory—to fine-tune Gemma models, each with its own unique features and capabilities.

1. Finetune_with_XTuner.ipynb

  • Toolkit: XTuner - An efficient, flexible, and full-featured toolkit specifically designed for fine-tuning LLMs.
  • Model: Gemma-2b-it
  • Dataset: Alpaca dataset

2. Finetune_with_Axolotl.ipynb

  • Toolkit: Axolotl - A tool designed to streamline the fine-tuning of various AI models, offering support for multiple configurations and architectures.  
  • Model: Gemma-2b
  • Dataset: mhenrichsen/alpaca_2k_test

3. Finetune_with_LLaMA_Factory.ipynb

Sign up or log in to comment