🏗️ Ko-Gemma-2-9b-Safety-FineTuned

1. 모델 소개 (Introduction)

건설 현장의 사고 예방을 위해 한국어 Gemma-2-9b 모델을 Fine-tuning한 안전 특화 언어 모델입니다. 건설 현장에서 발생할 수 있는 다양한 위험 상황에 대해 선제적인 안전 수칙과 대응 방안을 제시합니다.

2. 사용 예시 코드 (Quick Start)

from transformers import AutoModelForCausalLM, AutoTokenizer
import torch

# 모델 ID 설정
model_id = "lee124/Ko-Gemma-2-9b-Safety-FineTuned"

# 토크나이저 및 모델 로드
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(
    model_id,
    device_map="auto",
    torch_dtype=torch.bfloat16
)

# 입력 텍스트
input_text = "건설 현장에서 추락 사고를 예방하려면 어떻게 해야 해?"
inputs = tokenizer(input_text, return_tensors="pt").to("cuda")

# 텍스트 생성
outputs = model.generate(**inputs, max_new_tokens=256)
print(tokenizer.decode(outputs, skip_special_tokens=True))

3. 학습 정보 (Training Info)

● Base Model: rtzr/ko-gemma-2-9b-it

● Method: QLoRA (PEFT)

● Dataset: Construction Accident Prevention Data (Synthetic)

● License: Gemma