Add image benchmark action (#21)

fixie-ai · May 5, 2024 · 4657877 · 4657877
1 parent e734bb0
commit 4657877
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/.github/workflows/bench.yml b/.github/workflows/bench.yml
@@ -13,8 +13,10 @@ jobs:
     steps:
       - name: Send Benchmark Requests
         run: |
-          regions=("sea" "iad" "cdg")          
+          regions=("sea" "iad" "cdg")
+          media=("text" "image")        
           for region in "${regions[@]}"
+          for media in "${media[@]}"
           do
-            curl -f -X POST 'https://ai-benchmarks.fly.dev/bench?mode=text&max_tokens=20&spread=30&store' -H "fly-prefer-region: $region"          
+            curl -f -X POST "https://ai-benchmarks.fly.dev/bench?mode=$media&max_tokens=20&spread=30&store" -H "fly-prefer-region: $region"          
           done