Skip to content

Commit

Permalink
added vision config
Browse files Browse the repository at this point in the history
  • Loading branch information
lisadunlap committed Sep 25, 2024
1 parent 3a908ab commit 2a85597
Showing 1 changed file with 27 additions and 0 deletions.
27 changes: 27 additions & 0 deletions fastchat/serve/monitor/classify/vision_config.yaml
Original file line number Diff line number Diff line change
@@ -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$

0 comments on commit 2a85597

Please sign in to comment.