Update README.md
Browse files
README.md
CHANGED
|
@@ -28,7 +28,13 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
| 28 |
)
|
| 29 |
model.eval()
|
| 30 |
|
| 31 |
-
input_text = "
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 32 |
input_ids = tokenizer.encode(input_text, return_tensors="pt")
|
| 33 |
|
| 34 |
with torch.no_grad():
|
|
|
|
| 28 |
)
|
| 29 |
model.eval()
|
| 30 |
|
| 31 |
+
input_text = """# system
|
| 32 |
+
誠実で紳士的で優秀なAIアシスタントとして、簡潔でわかりやすく役に立つ回答を自信をもって答えなさい。
|
| 33 |
+
|
| 34 |
+
# question
|
| 35 |
+
人とAIが協調するためには？
|
| 36 |
+
|
| 37 |
+
# answer"""
|
| 38 |
input_ids = tokenizer.encode(input_text, return_tensors="pt")
|
| 39 |
|
| 40 |
with torch.no_grad():
|