llm-jp-3-13b-finetune-4-dpo-2

LLM-JP-3-13B ファインチューニングモデル

モデル詳細

ベースモデル: llm-jp/llm-jp-3-13b
アダプターモデル 1: nmczzi/llm-jp-3-13b-finetune-4
アダプターモデル 2: nmczzi/llm-jp-3-13b-finetune-4-dpo-2
量子化: 4ビット量子化 (QLoRA)

インストール

必要なパッケージのインストール：

pip install -U bitsandbytes transformers accelerate datasets peft

使用方法

以下は、モデルの基本的な使用例です(python)：

from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig from peft import PeftModel, LoraConfig import torch

HF_TOKEN = "有効なHuggingFaceトークン"

from google.colab import userdata HF_TOKEN = userdata.get('HF_API_KEY')

base_model_id = "llm-jp/llm-jp-3-13b" adapter_id_1 = "nmczzi/llm-jp-3-13b-finetune-4" adapter_id_2 = "nmczzi/llm-jp-3-13b-finetune-4-dpo-2"

QLoRA設定

bnb_config = BitsAndBytesConfig( load_in_4bit=True, bnb_4bit_quant_type="nf4", bnb_4bit_compute_dtype=torch.bfloat16, )

モデルの読み込み

model = AutoModelForCausalLM.from_pretrained( base_model_id, quantization_config=bnb_config, device_map="auto", token=HF_TOKEN )

トークナイザーの読み込み

tokenizer = AutoTokenizer.from_pretrained(base_model_id, trust_remote_code=True, token=HF_TOKEN)

ファインチューニングされたアダプターの読み込み

model = PeftModel.from_pretrained(model, adapter_id_1, token=HF_TOKEN) model = PeftModel.from_pretrained(model, adapter_id_2, token=HF_TOKEN)

生成用の関数を定義

def generate_response(input): prompt = f"""### 指示 {input} ### 回答 """

tokenized_input = tokenizer.encode(prompt, add_special_tokens=False, return_tensors="pt").to(model.device) attention_mask = torch.ones_like(tokenized_input)

with torch.no_grad(): outputs = model.generate( tokenized_input, attention_mask=attention_mask, max_new_tokens=100, do_sample=False, repetition_penalty=1.2, pad_token_id=tokenizer.eos_token_id )[0] output = tokenizer.decode(outputs[tokenized_input.size(1):], skip_special_tokens=True) return output

使用例

input = "### 指示\nあなたの指示をここに入力してください\n### 回答\n" inputs = tokenizer(prompt, return_tensors="pt").to(model.device)

with torch.no_grad(): outputs = model.generate(**inputs, max_new_tokens=100, do_sample=False, repetition_penalty=1.2)

response = tokenizer.decode(outputs[0], skip_special_tokens=True) print(response)