kazuHF commited on
Commit
153984d
·
verified ·
1 Parent(s): bbc800e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -2
README.md CHANGED
@@ -66,14 +66,15 @@ model, tokenizer = FastLanguageModel.from_pretrained(
66
 
67
  # 元のモデルにLoRAのアダプタを統合。
68
  model = PeftModel.from_pretrained(model, adapter_id, token = HF_TOKEN)
69
- # 推論するためにモデルのモードを変更
70
- FastLanguageModel.for_inference(model)
71
  ```
72
  3. 単一の入力文に対して推論する場合
73
 
74
  ```bash
75
  # 単一の入力文に基づいて推論する関数の定義。
76
  def Decoder(input):
 
 
 
77
  prompt = f"""### 指示\n\n{str(input)}\n\n### 回答"""
78
  inputs = tokenizer([prompt], return_tensors = "pt").to(model.device)
79
  outputs = model.generate(**inputs, max_new_tokens = 512, use_cache = True, do_sample=False, repetition_penalty=1.2)
@@ -104,6 +105,9 @@ with open("./elyza-tasks-100-TV_0.jsonl", "r") as f:
104
  # モデルで入力を一括処理。
105
  results = []
106
  for dt in tqdm(datasets):
 
 
 
107
  input = dt["input"]
108
  prompt = f"""### 指示\n{input}\n### 回答\n"""
109
  inputs = tokenizer([prompt], return_tensors = "pt").to(model.device)
 
66
 
67
  # 元のモデルにLoRAのアダプタを統合。
68
  model = PeftModel.from_pretrained(model, adapter_id, token = HF_TOKEN)
 
 
69
  ```
70
  3. 単一の入力文に対して推論する場合
71
 
72
  ```bash
73
  # 単一の入力文に基づいて推論する関数の定義。
74
  def Decoder(input):
75
+ # 推論するためにモデルのモードを変更
76
+ FastLanguageModel.for_inference(model)
77
+ # 入力文による推論
78
  prompt = f"""### 指示\n\n{str(input)}\n\n### 回答"""
79
  inputs = tokenizer([prompt], return_tensors = "pt").to(model.device)
80
  outputs = model.generate(**inputs, max_new_tokens = 512, use_cache = True, do_sample=False, repetition_penalty=1.2)
 
105
  # モデルで入力を一括処理。
106
  results = []
107
  for dt in tqdm(datasets):
108
+ # 推論するためにモデルのモードを変更
109
+ FastLanguageModel.for_inference(model)
110
+ # 入力文による推論
111
  input = dt["input"]
112
  prompt = f"""### 指示\n{input}\n### 回答\n"""
113
  inputs = tokenizer([prompt], return_tensors = "pt").to(model.device)