Edit fields

Signed-off-by: Thanawan Atchariyachanvanit <[email protected]>
thanawan-atc · Jul 7, 2023 · 243f800 · 243f800
1 parent 9f9c4c0
commit 243f800
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/.github/workflows/model-uploader.yml b/.github/workflows/model-uploader.yml
@@ -14,6 +14,7 @@ on:
         description: "Model format for auto-tracing"
         required: true
         type: choice
+        default: "Both"
         options:
         - "TORCH_SCRIPT"
         - "ONNX"
@@ -26,6 +27,12 @@ on:
         description: "(Optional) You can specify the pooling mode of the model here if it does not exist in original config.json file."
         required: false
         type: string
+        options:
+        - ""
+        - "CLS"
+        - "MEAN"
+        - "MAX"
+        - "MEAN_SQRT_LEN"
 
 jobs:
   model-tracing: