grug prompt compressor
Binary preserve/discard token classifier for prompt compression, trained with
grug train on microsoft/MeetingBank-LLMCompressed.
import grug
from grug.backends.modern import ModernBackend
comp = grug.Compressor(ModernBackend(model_name='akshayballal/grug-ettin-32m-meetingbank-v2'))
result = comp.compress(text, rate=0.4)
Training
{
"model_name": "jhu-clsp/ettin-encoder-32m",
"epochs": 6,
"learning_rate": 1e-05,
"batch_size": 32,
"max_length": 512,
"weight_decay": 0.01,
"warmup_ratio": 0.06,
"device": "cuda",
"seed": 0,
"push_to": "akshayballal/grug-ettin-32m-meetingbank-v2",
"select_on": "val_f1",
"patience": 2,
"cs_weight": 0.0
}
Final epoch
{
"epoch": 6,
"train_loss": 0.3438479603842051,
"val_loss": 0.3522194652062542,
"val_accuracy": 0.8408598789818473,
"val_precision": 0.7949736770080338,
"val_recall": 0.7951144890663747,
"val_f1": 0.7950440768023185,
"val_perplexity": 1.4222206174116105
}
The training corpus is CC-BY-NC-SA-4.0, so this model inherits a non-commercial constraint.