chatbot
Collection
4 items • Updated
How to use DuongTrongChi/gemma-2b-chatml with PEFT:
from peft import PeftModel
from transformers import AutoModelForCausalLM
base_model = AutoModelForCausalLM.from_pretrained("google/gemma-2b-it")
model = PeftModel.from_pretrained(base_model, "DuongTrongChi/gemma-2b-chatml")This model is a fine-tuned version of google/gemma-2b-it on the generator dataset.
More information needed
More information needed
More information needed
The following hyperparameters were used during training:
Base model
google/gemma-2b-it