|
--- |
|
library_name: transformers |
|
tags: [] |
|
--- |
|
|
|
## llava-jp-LoRA-1.3b-v1.1-Dog Model Card |
|
このモデルは[LLaVA-JP-13b-v1.1](https://huggingface.co/toshi456/llava-jp-1.3b-v1.1)からLoRAを作成し、画像に写っている犬の気持ちをなんとなくで言語化するモデルです。<br> |
|
精度は信頼できません。 |
|
|
|
## 使い方 |
|
当モデルは[densenet121-dog-emotions](https://huggingface.co/shinyice/densenet121-dog-emotions)と一緒に使うことを前提にしています。<br> |
|
プロンプトは |
|
```sh |
|
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() |
|
``` |
|
となっています。 |
