Update import model for MMDetection Models (#2586)

Azure · Aug 29, 2023 · 5fb2810 · 5fb2810
1 parent 1aa6932
commit 5fb2810
Show file tree

Hide file tree

Showing 5 changed files with 663 additions and 54 deletions.
diff --git a/.github/workflows/sdk-foundation-models-system-import-import_model_into_registry.yml b/.github/workflows/sdk-foundation-models-system-import-import_model_into_registry.yml
@@ -19,6 +19,9 @@ on:
       - sdk/python/dev-requirements.txt
       - infra/bootstrapping/**
       - sdk/python/setup.sh
+env:
+  MMDETECTION_MODEL_ID: None
+  MMDETECTION_TASK_NAME: image-object-detection
 concurrency:
   group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
   cancel-in-progress: true

diff --git a/...ub/workflows/sdk-foundation-models-system-import-import_model_into_registry_new_model.yml b/...ub/workflows/sdk-foundation-models-system-import-import_model_into_registry_new_model.yml
@@ -17,6 +17,8 @@ on:
 env:
   MODEL_ID: ai4bharat/indic-bert
   TASK_NAME: fill-mask
+  MMDETECTION_MODEL_ID: None
+  MMDETECTION_TASK_NAME: image-object-detection
 concurrency:
   group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
   cancel-in-progress: true

diff --git a/...gistry_new_model_image_classification.yml → ...l_into_registry_new_model_image_tasks.yml b/...gistry_new_model_image_classification.yml → ...l_into_registry_new_model_image_tasks.yml
@@ -1,4 +1,4 @@
-name: sdk-foundation-models-system-import-import_model_into_registry_new_model_image_classification
+name: sdk-foundation-models-system-import-import_model_into_registry_new_model_image_tasks
 # This file is created by sdk/python/readme.py.
 # Please do not edit directly.
 on:
@@ -10,13 +10,15 @@ on:
       - main
     paths:
       - sdk/python/foundation-models/system/import/**
-      - .github/workflows/sdk-foundation-models-system-import-import_model_into_registry_new_model_image_classification.yml
+      - .github/workflows/sdk-foundation-models-system-import-import_model_into_registry_new_model_image_tasks.yml
       - sdk/python/dev-requirements.txt
       - infra/bootstrapping/**
       - sdk/python/setup.sh
 env:
   MODEL_ID: microsoft/resnet-18
   TASK_NAME: image-classification
+  MMDETECTION_MODEL_ID: faster_rcnn_r50_fpn_1x_coco
+  MMDETECTION_TASK_NAME: image-object-detection
 concurrency:
   group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
   cancel-in-progress: true