AlicanKiraz0/Cybersecurity-Dataset-Fenrir-v2.1
Viewer • Updated • 99.9k • 3.99k • 122
How to use ygzgrdmr/gemma-4-31B-fenrir-cybersec-hpe-dl384 with PEFT:
from peft import PeftModel
from transformers import AutoModelForCausalLM
base_model = AutoModelForCausalLM.from_pretrained("google/gemma-4-31B")
model = PeftModel.from_pretrained(base_model, "ygzgrdmr/gemma-4-31B-fenrir-cybersec-hpe-dl384")LoRA fine-tune of google/gemma-4-31B on
AlicanKiraz0/Cybersecurity-Dataset-Fenrir-v2.1
by Alican Kiraz, trained on an HPE ProLiant Compute DL384 Gen12 (2× NVIDIA GH200 144GB HBM3e).
AlicanKiraz0/Cybersecurity-Dataset-Fenrir-v2.1 (99,870 rows; system / user / assistant turns){
"train_runtime": 145390.0266,
"train_samples_per_second": 2.061,
"train_steps_per_second": 0.129,
"total_flos": 6.327548262623268e+19,
"train_loss": 0.4692629598424079
}
from peft import PeftModel
from transformers import AutoModelForImageTextToText, AutoTokenizer
base = AutoModelForImageTextToText.from_pretrained("google/gemma-4-31B", torch_dtype="bfloat16")
model = PeftModel.from_pretrained(base, "ygzgrdmr/gemma-4-31B-fenrir-cybersec-hpe-dl384")
tok = AutoTokenizer.from_pretrained("google/gemma-4-31B")
Base model
google/gemma-4-31B