mizydorczyk/gemma-4-e4b-it-ask-dataset
Viewer • Updated • 177 • 24
How to use mizydorczyk/gemma-4-e4b-it-ask-lora with PEFT:
from peft import PeftModel
from transformers import AutoModelForCausalLM
base_model = AutoModelForCausalLM.from_pretrained("google/gemma-4-E4B-it")
model = PeftModel.from_pretrained(base_model, "mizydorczyk/gemma-4-e4b-it-ask-lora")ask text-only LoRA adapter for Gemma4ForCausalLM.
| Setting | Value |
|---|---|
| Train examples | 150 |
| Evaluate examples | 27 |
| Base architecture | Gemma4ForCausalLM (text-only examples) |
| LoRA targets | All linear layers in the text model |
| LoRA rank | 8 |
| LoRA alpha | 32 |
| LoRA dropout | 0.05 |
| Precision | BF16 |
| Maximum sequence length | 4096 |
| Effective batch size | 16 |
| Epochs | 5 |
| Learning rate | 0.0001 |
| Warmup steps | 5 |
| Selected checkpoint step | 30 |
Selected eval_loss |
0.6459568738937378 |
| Loss | Assistant-only |
| Metric | Value |
|---|---|
train_runtime |
827.6625 |
train_samples_per_second |
0.906 |
train_steps_per_second |
0.06 |
total_flos |
4371747958794240.0 |
train_loss |
0.8483663254976272 |
epoch |
5.0 |
eval_loss |
0.6459568738937378 |
eval_entropy |
0.5192408040165901 |
eval_num_tokens |
182485.0 |
eval_mean_token_accuracy |
0.8542580008506775 |
| Source | Repository | Revision |
|---|---|---|
| Base model | google/gemma-4-E4B-it |
ee0ef6023621cff504d758262d4e04895a5af4a2 |
| Dataset | mizydorczyk/gemma-4-e4b-it-ask-dataset |
main |