Lamini Prompt Enchance Long

🚀 Lamini-Prompt-Enchance-Long

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

• 損失: 2.1624
• Rouge1: 20.2443
• Rouge2: 9.3642
• Rougel: 17.2484
• Rougelsum: 19.0703
• 生成長: 19.0

🚀 クイックスタート

以下はこのモデルを使用するためのコード例です。

from transformers import pipeline     

# load model and tokenizer from huggingface hub with pipeline
enhancer = pipeline("summarization", model="gokaygokay/Lamini-Prompt-Enchance-Long", 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'])
 

💻 使用例

基本的な使用法

from transformers import pipeline     

# load model and tokenizer from huggingface hub with pipeline
enhancer = pipeline("summarization", model="gokaygokay/Lamini-Prompt-Enchance-Long", 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'])

🔧 技術詳細

学習ハイパーパラメータ

学習時には以下のハイパーパラメータが使用されました。

• 学習率: 5e-05
• 学習バッチサイズ: 8
• 評価バッチサイズ: 8
• シード: 42
• オプティマイザ: Adam with betas=(0.9,0.999) and epsilon=1e-08
• 学習率スケジューラの種類: linear
• エポック数: 5

学習結果

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

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

📄 ライセンス

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