aripos1/gorani_dataset
Viewer • Updated • 27.1k • 22
How to use lucian1120/gorani-8b with PEFT:
from peft import PeftModel
from transformers import AutoModelForCausalLM
base_model = AutoModelForCausalLM.from_pretrained("unsloth/Meta-Llama-3.1-8B-Instruct-bnb-4bit")
model = PeftModel.from_pretrained(base_model, "lucian1120/gorani-8b")이 모델은 번역을 위한 모델입니다. 한국 고유어의 정확한 번역을 생성하기 위해 한국어, 영어, 일본어의 언어 데이터를 혼합하여 unsloth/Llama-3.1-8B-Instruct-bnb-4bit을 학습시켜 생성된 gorani-8B 입니다.
gorani는 현재 한국어, 영어, 일본어만 번역을 지원합니다.
not is_bfloat16_supported() ![점수 비교 그래프]
| 평가 지표 | LLama 3.1-8B | Gorani-8B |
|---|---|---|
| BERTScore (Precision) | 0.7123 | 0.8456 |
| BERTScore (Recall) | 0.6987 | 0.8621 |
| BERTScore (F1 Score) | 0.7055 | 0.8532 |
| COMET Score | 0.5902 | 0.8123 |
Base model
meta-llama/Llama-3.1-8B