Skip to content

Commit 19be58c

Browse files
committed
added vision config
1 parent a5848a9 commit 19be58c

File tree

1 file changed

+34
-0
lines changed

1 file changed

+34
-0
lines changed
Lines changed: 34 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,34 @@
1+
# Yaml config file for category classification
2+
3+
input_file: null # json
4+
cache_file: null # json
5+
output_file: null # json line
6+
7+
convert_to_json: True
8+
9+
task_name:
10+
- captioning_v0.1
11+
- homework_v0.1
12+
- ocr_v0.1
13+
- humor_v0.1
14+
- entity_recognition_v0.1
15+
- creative_writing_vision_v0.1
16+
- diagram_v0.1
17+
18+
19+
model_name: null
20+
name: gemini-1.5-flash
21+
api_type: gemini
22+
endpoints:
23+
- api_base: null
24+
api_key: null
25+
26+
parallel: 50
27+
temperature: 0.0
28+
max_token: 512
29+
30+
image_dir: null # directory where vision arena images are stored
31+
32+
max_retry: 2
33+
retry_sleep: 10
34+
error_output: $ERROR$

0 commit comments

Comments
 (0)