lightnovel-brush-up-lora

Light Novel Brush-up LoRA

ライトノベルのテキスト改善に特化したLoRAアダプターです。

概要

このモデルは、ライトノベルの文章を改善するために学習されたLoRAアダプターです。

改善パターン

• 羅列→実感変換：単語の羅列を自然な文章に
• 擬音の最適化：擬音を3-5個に厳選
• 冗長表現削減：冗長な表現を簡潔に
• 人名主語削減：連続する人名主語を自然に
• キャラ歯車化防止：機械的な動作を感情的に
• 健忘症キャラ防止：既知情報の説明削除

学習データ

• データ数: 603サンプル
• エポック数: 2.96
• 最終損失: 1.265

使用方法

Pythonスクリプト

from transformers import AutoModelForCausalLM, AutoTokenizer
from peft import PeftModel

BASE_MODEL = "elyza/Llama-3-ELYZA-JP-8B"
LORA_MODEL = "yozora4416/lightnovel-brush-up-lora"

tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL)
model = AutoModelForCausalLM.from_pretrained(BASE_MODEL, load_in_8bit=True)
model = PeftModel.from_pretrained(model, LORA_MODEL)

prompt = """### 指示:
以下の文章を改善してください。

### 入力:
振り返る。敵。距離は近い。

### 出力:
"""

inputs = tokenizer(prompt, return_tensors="pt")
outputs = model.generate(**inputs, max_new_tokens=256)
print(tokenizer.decode(outputs[0]))

ライセンス

Apache 2.0

作者

学習済みLoRAアダプター