gemma-7B-it-firefly

Model Card for Firefly-Gemma

gemma-7B-it-firefly is trained based on gemma-7b-it to act as a helpful and harmless AI assistant. we trained the model on firefly-train-1.1M dataset using LoRA.

<img src="gemma-7B-it-firefly.jpg" width="250">

Performance

we evaluate the model on Open LLM Leaderboard

Usage

The chat template of our chat models is same as Official gemma-7b-it:

<bos><start_of_turn>user
Write a hello world program<end_of_turn>
<start_of_turn>model

You can also use the following code:

from transformers import AutoTokenizer, AutoModelForCausalLM

model_name_or_path = "yys/gemma-7B-it-firefly"
tokenizer = AutoTokenizer.from_pretrained(model_name_or_path)
model = AutoModelForCausalLM.from_pretrained(model_name_or_path)

input_text = "给我写一首关于机器学习的诗歌。"
input_ids = tokenizer(input_text, return_tensors="pt")

outputs = model.generate(**input_ids)
print(tokenizer.decode(outputs[0]))