Update README.md
Browse files
README.md
CHANGED
@@ -19,7 +19,7 @@ This model is released under the Apache License 2.0.
|
|
19 |
## Usage
|
20 |
Install the required libraries as follows:
|
21 |
```sh
|
22 |
-
>>> python -m pip install numpy
|
23 |
|
24 |
```
|
25 |
|
@@ -56,7 +56,7 @@ def generate_prompt(messages: list) -> str:
|
|
56 |
"以下はタスクを説明する指示で、文脈を説明した入力とペアになっています。",
|
57 |
"要求を適切に補完するよう応答を書いてください。",
|
58 |
]
|
59 |
-
roles = {"instruction": "指示",
|
60 |
for msg in messages:
|
61 |
prompt.append(sep + roles[msg["role"]] + ":\n" + msg['content'])
|
62 |
prompt.append(sep + roles["response"] + ":\n")
|
|
|
19 |
## Usage
|
20 |
Install the required libraries as follows:
|
21 |
```sh
|
22 |
+
>>> python -m pip install numpy sentencepiece torch transformers
|
23 |
|
24 |
```
|
25 |
|
|
|
56 |
"以下はタスクを説明する指示で、文脈を説明した入力とペアになっています。",
|
57 |
"要求を適切に補完するよう応答を書いてください。",
|
58 |
]
|
59 |
+
roles = {"instruction": "指示", "response": "応答", "input": "入力"}
|
60 |
for msg in messages:
|
61 |
prompt.append(sep + roles[msg["role"]] + ":\n" + msg['content'])
|
62 |
prompt.append(sep + roles["response"] + ":\n")
|