Dedicated Endpoints

Run this model inference on single tenant GPU with unmatched speed and reliability at scale.

Learn more
Container

Run this model inference with full control and performance in your environment.

Learn more

Get help setting up a custom Dedicated Endpoints.

Talk with our engineer to get a quote for reserved GPU instances with discounts.

README

License: apache-2.0

Base model

google/gemma-2-2b-it

Description

QLoRA adapter — Gemma-2-2B-IT fine-tuned on NLNG LNG domain Q&A

Training data covers NLNG's internal domain: gas supply chain, cargo scheduling, shipping logistics, LNG production, HSE compliance, thermodynamics, and commercial operations.

Usage

python

from peft import PeftModel
from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
import torch
bnb_cfg = BitsAndBytesConfig(load_in_4bit=True, bnb_4bit_quant_type="nf4",
bnb_4bit_compute_dtype=torch.bfloat16)
tokenizer = AutoTokenizer.from_pretrained("google/gemma-2-2b-it")
base = AutoModelForCausalLM.from_pretrained(
"google/gemma-2-2b-it", quantization_config=bnb_cfg, device_map="auto")
model = PeftModel.from_pretrained(base, "Bluechip-AI/nlng-slm-v3-gemma")
prompt = "What are the key safety considerations for LNG cargo loading operations?"
inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
output = model.generate(**inputs, max_new_tokens=256)
print(tokenizer.decode(output[0], skip_special_tokens=True))

Training details

ParameterValue
LoRA r16
LoRA alpha32
LoRA dropout0.05
Quantisation4-bit nf4
Epochs3
Batch size2 (grad accum 4)
Learning rate2e-4
LR schedulecosine

Model provider

Bluechip-AI

Model tree

Base

google/gemma-2-2b-it

Adapter

this model

Modalities

Input

Text

Output

Text

Pricing

Dedicated Endpoints

View details

Supported Functionality

Model APIs

Dedicated Endpoints

Container

More information

Explore FriendliAI today