hiroki-rad
commited on
Update README.md
Browse files
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 |
|