Update README.md
Browse files
README.md
CHANGED
|
@@ -49,9 +49,12 @@ For more details, please refer to [our blog post]().
|
|
| 49 |
|
| 50 |
## Usage
|
| 51 |
```python
|
|
|
|
| 52 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 53 |
from peft import PeftModel
|
| 54 |
|
|
|
|
|
|
|
| 55 |
PROMPT = """### 指示:
|
| 56 |
あなたは優秀なアシスタントです。ユーザークエリが下記のカテゴリに当てはまるか判定し、当てはまる場合は'回答拒否'と判定し、その後に関連するカテゴリを出力してください。当てはまらない場合は'回答許可'と判定しカテゴリは'無し'としてください。ただし、当てはまらない場合でも倫理的に回答してはいけないと判断した場合は'回答拒否'と判定しカテゴリは'その他'としてください。
|
| 57 |
|
|
|
|
| 49 |
|
| 50 |
## Usage
|
| 51 |
```python
|
| 52 |
+
from huggingface_hub import login
|
| 53 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 54 |
from peft import PeftModel
|
| 55 |
|
| 56 |
+
login(token="your_huggingface_token")
|
| 57 |
+
|
| 58 |
PROMPT = """### 指示:
|
| 59 |
あなたは優秀なアシスタントです。ユーザークエリが下記のカテゴリに当てはまるか判定し、当てはまる場合は'回答拒否'と判定し、その後に関連するカテゴリを出力してください。当てはまらない場合は'回答許可'と判定しカテゴリは'無し'としてください。ただし、当てはまらない場合でも倫理的に回答してはいけないと判断した場合は'回答拒否'と判定しカテゴリは'その他'としてください。
|
| 60 |
|