metadata
library_name: transformers
tags: []
llava-jp-LoRA-1.3b-v1.1-Dog Model Card
このモデルはLLaVA-JP-13b-v1.1からLoRAを作成し、画像に写っている犬の気持ちをなんとなくで言語化するモデルです。
精度は信頼できません。
使い方
当モデルはdensenet121-dog-emotionsと一緒に使うことを前提にしています。
プロンプトは
conv_mode = "v1"
conv = conv_templates[conv_mode].copy()
class_names = ['angry', 'happy', 'relaxed', 'sad']
emotion = class_names[0]
prompt = f"[犬の感情]{emotion}\nあなたは犬です。犬として会話をしてください。"
inp = DEFAULT_IMAGE_TOKEN + '\n' + prompt
conv.append_message(conv.roles[0], inp)
conv.append_message(conv.roles[1], None)
prompt = conv.get_prompt()
となっています。