hiroki-rad commited on
Commit
55aa8d1
·
verified ·
1 Parent(s): f8090e0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -1
README.md CHANGED
@@ -21,7 +21,8 @@ import pandas as pd
21
  from vllm import LLM
22
  from tqdm import tqdm
23
  import json
24
- from datasets import
 
25
 
26
  from custom_few_shot_prompt_template import CustomFewShotPromptTemplate
27
 
@@ -65,6 +66,7 @@ thinking_generator = TwoStageThinking(llm)
65
  # 最終的に1回推論の回答を使うことにした
66
  results = []
67
  for row, few_shot in tqdm(zip(data.itertuples(), few_shot_list), desc="生成中"):
 
68
  first = thinking_generator.generate_complete_response(row, few_shot)
69
  results.append(first)
70
 
 
21
  from vllm import LLM
22
  from tqdm import tqdm
23
  import json
24
+ from datasets import
25
+ import time
26
 
27
  from custom_few_shot_prompt_template import CustomFewShotPromptTemplate
28
 
 
66
  # 最終的に1回推論の回答を使うことにした
67
  results = []
68
  for row, few_shot in tqdm(zip(data.itertuples(), few_shot_list), desc="生成中"):
69
+ time.sleep(3)
70
  first = thinking_generator.generate_complete_response(row, few_shot)
71
  results.append(first)
72