From 2a85597f4fcaf82ac1299e8ebf1c69275bd48f17 Mon Sep 17 00:00:00 2001
From: lisadunlap <lisabdunlap@gmail.com>
Date: Wed, 25 Sep 2024 23:03:52 +0000
Subject: [PATCH] added vision config

---
 .../serve/monitor/classify/vision_config.yaml | 27 +++++++++++++++++++
 1 file changed, 27 insertions(+)
 create mode 100644 fastchat/serve/monitor/classify/vision_config.yaml

diff --git a/fastchat/serve/monitor/classify/vision_config.yaml b/fastchat/serve/monitor/classify/vision_config.yaml
new file mode 100644
index 000000000..a7890c0e9
--- /dev/null
+++ b/fastchat/serve/monitor/classify/vision_config.yaml
@@ -0,0 +1,27 @@
+# Yaml config file for category classification
+
+input_file: "../arena-data-analysis/data/vision_clean_battle_conv_20240822_with_image_hash.json" # json
+cache_file: False # json
+output_file: "fastchat/serve/monitor/classify/results/vision_clean_battle_conv_20240822_with_image_hash-labeled.json" # json line
+
+convert_to_json: True
+
+task_name:
+  - refusal_v0.1
+  - criteria_v0.1
+  - if_v0.1
+  - vision_v0.1
+
+model_name: gpt-4o
+name: gpt-4o
+endpoints:
+  - api_base: # BASE URL
+    api_key: # API KEY
+
+parallel: 50
+temperature: 0.0
+max_token: 512
+
+max_retry: 2
+retry_sleep: 10
+error_output: $ERROR$
\ No newline at end of file