chimbiwide's picture
Update README.md
a112570 verified
metadata
base_model: unsloth/gemma-3n-e4b-it-unsloth-bnb-4bit
tags:
  - text-generation-inference
  - transformers
  - unsloth
  - gemma3n
license: apache-2.0
language:
  - en
datasets:
  - chimbiwide/RolePlay-NPC
pipeline_tag: image-text-to-text

Gemma3NPC-it

The "flagship" Gemma3NPC model that delivers good NPC role-playing at great speed.

We trained this model as a rank-16 LoRA adapter with one epoch over RolePlay-NPC using a 40GB vRAM A100 in Google Colab. For this run, we employed a learning rate of 2e-5 and a total batch size of 1 and gradient accumulation steps of 16. A cosine learning rate scheduler was used with an 800-step warmup. With a gradient clipping of 0.4.

Check out our training notebook here.


Here is a graph of the Step Training Loss, saved every 10 steps:

image/png