Update README.md
Browse files
README.md
CHANGED
@@ -17,5 +17,10 @@ conv = conv_templates[conv_mode].copy()
|
|
17 |
class_names = ['angry', 'happy', 'relaxed', 'sad']
|
18 |
emotion = class_names[0]
|
19 |
prompt = f"[犬の感情]{emotion}\nあなたは犬です。犬として会話をしてください。"
|
|
|
|
|
|
|
|
|
|
|
20 |
```
|
21 |
となっています。
|
|
|
17 |
class_names = ['angry', 'happy', 'relaxed', 'sad']
|
18 |
emotion = class_names[0]
|
19 |
prompt = f"[犬の感情]{emotion}\nあなたは犬です。犬として会話をしてください。"
|
20 |
+
|
21 |
+
inp = DEFAULT_IMAGE_TOKEN + '\n' + prompt
|
22 |
+
conv.append_message(conv.roles[0], inp)
|
23 |
+
conv.append_message(conv.roles[1], None)
|
24 |
+
prompt = conv.get_prompt()
|
25 |
```
|
26 |
となっています。
|