gnumanth/gita
Viewer • Updated • 701 • 19 • 1
How to use gnumanth/gemma-gita with PEFT:
from peft import PeftModel
from transformers import AutoModelForCausalLM
base_model = AutoModelForCausalLM.from_pretrained("google/gemma-7b")
model = PeftModel.from_pretrained(base_model, "gnumanth/gemma-gita")This model is a fine-tuned version of google/gemma-7b on gnumanth/gita dataset.
Experimental fine tune gemma-7b model
Gita interpretation, no limitations.
SFTTrainer
The following hyperparameters were used during training:
TrainOutput(global_step=10, training_loss=0.5232708215713501, metrics={'train_runtime': 13.8846, 'train_samples_per_second': 2.881, 'train_steps_per_second': 0.72, 'total_flos': 55989006336000.0, 'train_loss': 0.5232708215713501, 'epoch': 10.0})
Base model
google/gemma-7b