Instructions to use rudalson/Llama-3.2-3B-Instruct-Legal-Chatbot with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

Libraries

How to use rudalson/Llama-3.2-3B-Instruct-Legal-Chatbot with Transformers:

# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-generation", model="rudalson/Llama-3.2-3B-Instruct-Legal-Chatbot")
messages = [
    {"role": "user", "content": "Who are you?"},
]
pipe(messages)

# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("rudalson/Llama-3.2-3B-Instruct-Legal-Chatbot")
model = AutoModelForCausalLM.from_pretrained("rudalson/Llama-3.2-3B-Instruct-Legal-Chatbot")
messages = [
    {"role": "user", "content": "Who are you?"},
]
inputs = tokenizer.apply_chat_template(
	messages,
	add_generation_prompt=True,
	tokenize=True,
	return_dict=True,
	return_tensors="pt",
).to(model.device)

outputs = model.generate(**inputs, max_new_tokens=40)
print(tokenizer.decode(outputs[0][inputs["input_ids"].shape[-1]:]))

Notebooks
Google Colab
Kaggle
Local Apps

vLLM

How to use rudalson/Llama-3.2-3B-Instruct-Legal-Chatbot with vLLM:

Install from pip and serve model

# Install vLLM from pip:
pip install vllm
# Start the vLLM server:
vllm serve "rudalson/Llama-3.2-3B-Instruct-Legal-Chatbot"
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:8000/v1/chat/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "rudalson/Llama-3.2-3B-Instruct-Legal-Chatbot",
		"messages": [
			{
				"role": "user",
				"content": "What is the capital of France?"
			}
		]
	}'

Use Docker

docker model run hf.co/rudalson/Llama-3.2-3B-Instruct-Legal-Chatbot

SGLang

How to use rudalson/Llama-3.2-3B-Instruct-Legal-Chatbot with SGLang:

Install from pip and serve model

# Install SGLang from pip:
pip install sglang
# Start the SGLang server:
python3 -m sglang.launch_server \
    --model-path "rudalson/Llama-3.2-3B-Instruct-Legal-Chatbot" \
    --host 0.0.0.0 \
    --port 30000
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:30000/v1/chat/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "rudalson/Llama-3.2-3B-Instruct-Legal-Chatbot",
		"messages": [
			{
				"role": "user",
				"content": "What is the capital of France?"
			}
		]
	}'

Use Docker images

docker run --gpus all \
    --shm-size 32g \
    -p 30000:30000 \
    -v ~/.cache/huggingface:/root/.cache/huggingface \
    --env "HF_TOKEN=<secret>" \
    --ipc=host \
    lmsysorg/sglang:latest \
    python3 -m sglang.launch_server \
        --model-path "rudalson/Llama-3.2-3B-Instruct-Legal-Chatbot" \
        --host 0.0.0.0 \
        --port 30000
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:30000/v1/chat/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "rudalson/Llama-3.2-3B-Instruct-Legal-Chatbot",
		"messages": [
			{
				"role": "user",
				"content": "What is the capital of France?"
			}
		]
	}'

Docker Model Runner
How to use rudalson/Llama-3.2-3B-Instruct-Legal-Chatbot with Docker Model Runner:
```
docker model run hf.co/rudalson/Llama-3.2-3B-Instruct-Legal-Chatbot
```

Llama-3.2-3B-Instruct-Legal-Chatbot

Model Description

이 모델은 meta-llama/Llama-3.2-3B-Instruct를 기반으로 한국 민사법 도메인 지식에 특화되도록 Fine-tuning(LoRA)된 모델입니다. 사용자의 법률적 질의에 대해 전문적인 답변을 제공하며, 관련 법령, 판례, 관할 법원 등의 법률 개체(Legal Entities)를 추출하여 함께 제시할 수 있도록 학습되었습니다.

Base Model: meta-llama/Llama-3.2-3B-Instruct
Language: Korean (ko)
Task: Question Answering, Text Generation, Legal Entity Extraction
Training Method: Supervised Fine-Tuning (SFT) with PEFT(LoRA)

Intended Use & Limitations

Intended Use

한국 민사법 관련 기초적인 법률 질의응답
법률 문서 요약 및 주요 법령/판례 번호 추출
법률 AI 어시스턴트 연구 및 교육용 레퍼런스

Limitations & Ethical Considerations

주의사항: 이 모델은 교육 및 연구 목적으로 개발되었으며, 전문적인 법률 상담이나 변호사를 대체할 수 없습니다. * LLM의 특성상 환각(Hallucination) 현상이 발생하여 존재하지 않는 법령이나 판례를 생성할 수 있습니다. 실제 법적 조치가 필요한 경우 반드시 법률 전문가(변호사 등)의 조언을 구해야 합니다.

Training Details

Training Data

출처: AI Hub (한국 민사법 질의응답 및 판례 라벨링 데이터)
- 질의응답 카테고리 75,624건
- rudalson/legal-qa-1k-dataset - 테스트 용으로 샘플링 데이터셋
전처리: 조항 번호, 날짜, 금액 형식 등을 정규화하였으며, Llama 3의 Chat Template(system-user-assistant) 구조로 변환하여 학습을 진행했습니다.

Training Procedure & Hyperparameters

모델 학습은 Hugging Face의 peft, trl (SFTTrainer) 라이브러리를 사용하여 진행되었습니다.

LoRA Parameters:
- r: 16
- lora_alpha: 32
- target_modules: q_proj, k_proj, v_proj, o_proj, gate_proj, up_proj, down_proj
- lora_dropout: 0.05
Training Hyperparameters:
- learning_rate: 2e-4
- num_train_epochs: 1
- per_device_train_batch_size: 4
- gradient_accumulation_steps: 4
- optimizer: adamw_torch
- fp16 / bfloat16: Enabled

Evaluation

평가는 ROUGE 스코어 및 법률 개체명(법령, 판례, 법원 등) 추출 정확도를 기반으로 수행되었습니다.

ROUGE-1: 0.2222
ROUGE-L: 0.2222

How to Get Started with the Model

import torch
from transformers import AutoModelForCausalLM, AutoTokenizer

# 허깅페이스에 업로드한 모델 경로
model_name = "rudalson/Llama-3.2-3B-Instruct-Legal-Chatbot"

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

# 2. 추론 테스트
prompt = "계약 해제 시 손해배상을 청구할 수 있나요?"
messages = [
    {"role": "system", "content": "당신은 한국 법률 전문가 AI 어시스턴트입니다. 사용자의 질문에 대해 정확하고 전문적인 답변을 제공하세요."},
    {"role": "user", "content": prompt}
]

# 채팅 템플릿 적용
formatted_prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
inputs = tokenizer(formatted_prompt, return_tensors="pt").to(model.device)

# 답변 생성
with torch.no_grad():
    outputs = model.generate(
        **inputs,
        max_new_tokens=512,
        temperature=0.7,
        top_p=0.9,
        repetition_penalty=1.1,
        do_sample=True
    )

response = tokenizer.decode(outputs[0][inputs['input_ids'].shape[1]:], skip_special_tokens=True)
print(f"질문: {prompt}")
print(f"답변: {response.strip()}")

Downloads last month: 165

Safetensors

Model size

3B params

Tensor type

F32

BF16

Model tree for rudalson/Llama-3.2-3B-Instruct-Legal-Chatbot

Base model

meta-llama/Llama-3.2-3B-Instruct

Finetuned

(1604)

this model