shibu-phys commited on
Commit
9fd3494
·
verified ·
1 Parent(s): e9dcb51

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -0
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