test: CI convert gemma2

janhq · Aug 11, 2024 · 8c3392b · 8c3392b
1 parent 3d3e033
commit 8c3392b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/.github/workflows/convert-model.yml b/.github/workflows/convert-model.yml
@@ -25,9 +25,9 @@ on:
 
 env:
   USER_NAME: cortexso
-  SOURCE_MODEL_ID: meta-llama/Meta-Llama-3.1-8B-Instruct #${{ inputs.source_model_id }}
-  SOURCE_MODEL_SIZE: 8b #${{ inputs.source_model_size }}
-  TARGET_MODEL_ID: llama3.1 #${{ inputs.target_model_id }}
+  SOURCE_MODEL_ID: google/gemma-2-2b-it #${{ inputs.source_model_id }}
+  SOURCE_MODEL_SIZE: 2b #${{ inputs.source_model_size }}
+  TARGET_MODEL_ID: gemma2 #${{ inputs.target_model_id }}
   PRECISION: int4                   # Valid values: int4,fp16,fp3
   EXECUTOR: dml                     # Valid values: cpu,cuda,dml,web
   ONNXRUNTIME_GENAI_VERSION: 0.3.0  # Check version from: https://github.com/microsoft/onnxruntime-genai/releases