Lamini Prompt Enchance

🚀 Lamini-Prompt-Enchance

このモデルは、未知のデータセットでMBZUAI/LaMini-Flan-T5-248Mをファインチューニングしたバージョンです。評価セットでは以下の結果を達成しています。

• 損失: 2.0195
• Rouge1: 31.5042
• Rouge2: 13.2633
• Rougel: 26.4176
• Rougelsum: 28.4846
• 生成長: 19.0

🚀 クイックスタート

💻 使用例

基本的な使用法

from transformers import pipeline     

# load model and tokenizer from huggingface hub with pipeline
enhancer = pipeline("summarization", model="gokaygokay/Lamini-Prompt-Enchance", device=0)

prompt = "A blue-tinted bedroom scene, surreal and serene, with a mysterious reflected interior."
prefix = "Enhance the description: "
# enhance prompt
res = enhancer(prefix + prompt)

print(res[0]['summary_text'])

# A surreal and serene bedroom scene with a mysterious mirrored interior, 
# awash in blue and green hues.
# The room is adorned with intricate patterns and a mirrored wall, 
# creating a sense of mystery and tranquility.

📦 インストール

このセクションでは、必要なライブラリのインストール方法を説明します。

pip install transformers

📚 ドキュメント

モデル情報

トレーニング手順

トレーニングハイパーパラメータ

トレーニング中に使用されたハイパーパラメータは以下の通りです。

• 学習率: 5e-05
• トレーニングバッチサイズ: 24
• 評価バッチサイズ: 8
• シード: 42
• オプティマイザ: Adam (betas=(0.9,0.999), epsilon=1e-08)
• 学習率スケジューラタイプ: 線形
• エポック数: 5

トレーニング結果

フレームワークバージョン

• Transformers 4.42.3
• Pytorch 2.3.0+cu121
• Datasets 2.20.0
• Tokenizers 0.19.1

📄 ライセンス

このモデルはCC BY-NC 4.0ライセンスの下で提供されています。