chore: misc
Browse files- autotab.py +9 -6
autotab.py
CHANGED
@@ -10,23 +10,26 @@ class AutoTab:
|
|
10 |
self,
|
11 |
in_file_path: str,
|
12 |
out_file_path: str,
|
|
|
13 |
max_examples: int,
|
14 |
model_name: str,
|
15 |
-
api_key: str,
|
16 |
-
base_url: str,
|
17 |
generation_config: dict,
|
|
|
18 |
save_every: int,
|
19 |
-
|
|
|
|
|
20 |
):
|
21 |
self.in_file_path = in_file_path
|
22 |
self.out_file_path = out_file_path
|
|
|
23 |
self.max_examples = max_examples
|
24 |
self.model_name = model_name
|
25 |
-
self.api_key = api_key
|
26 |
-
self.base_url = base_url
|
27 |
self.generation_config = generation_config
|
|
|
28 |
self.save_every = save_every
|
29 |
-
self.
|
|
|
30 |
|
31 |
# βββ IO βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
32 |
|
|
|
10 |
self,
|
11 |
in_file_path: str,
|
12 |
out_file_path: str,
|
13 |
+
instruction: str,
|
14 |
max_examples: int,
|
15 |
model_name: str,
|
|
|
|
|
16 |
generation_config: dict,
|
17 |
+
request_interval: float,
|
18 |
save_every: int,
|
19 |
+
api_key: str,
|
20 |
+
base_url: str,
|
21 |
+
|
22 |
):
|
23 |
self.in_file_path = in_file_path
|
24 |
self.out_file_path = out_file_path
|
25 |
+
self.instruction = instruction
|
26 |
self.max_examples = max_examples
|
27 |
self.model_name = model_name
|
|
|
|
|
28 |
self.generation_config = generation_config
|
29 |
+
self.request_interval = request_interval
|
30 |
self.save_every = save_every
|
31 |
+
self.api_key = api_key
|
32 |
+
self.base_url = base_url
|
33 |
|
34 |
# βββ IO βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
35 |
|