diff --git a/assets/training/model_management/components/convert_model_to_mlflow/spec.yaml b/assets/training/model_management/components/convert_model_to_mlflow/spec.yaml
index 96433d90ac..7d2a12753c 100644
--- a/assets/training/model_management/components/convert_model_to_mlflow/spec.yaml
+++ b/assets/training/model_management/components/convert_model_to_mlflow/spec.yaml
@@ -1,7 +1,7 @@
 $schema: https://azuremlschemas.azureedge.net/latest/commandComponent.schema.json
 
 name: convert_model_to_mlflow
-version: 0.0.32
+version: 0.0.33
 type: command
 
 is_deterministic: True
diff --git a/assets/training/model_management/components/import_model/spec.yaml b/assets/training/model_management/components/import_model/spec.yaml
index 62179e1926..2ff88ce5f8 100644
--- a/assets/training/model_management/components/import_model/spec.yaml
+++ b/assets/training/model_management/components/import_model/spec.yaml
@@ -4,7 +4,7 @@ type: pipeline
 name: import_model
 display_name: Import model
 description: Import a model into a workspace or a registry
-version: 0.0.38
+version: 0.0.39
 
 # Pipeline inputs
 inputs:
@@ -47,6 +47,17 @@ inputs:
     optional: false
     description: Flavor of MLFlow to which model the model is converted to.
 
+  model_framework:
+    type: string
+    enum:
+      - Huggingface
+      - MMLab
+      - llava
+      - AutoML
+    default: Huggingface
+    optional: false
+    description: Framework from which model is imported from.
+
   vllm_enabled:
     type: boolean
     description: Enable vllm in the converted model 
@@ -260,7 +271,7 @@ jobs:
         type: uri_folder
 
   convert_model_to_mlflow:
-    component: azureml:convert_model_to_mlflow:0.0.32
+    component: azureml:convert_model_to_mlflow:0.0.33
     compute: ${{parent.inputs.compute}}
     resources:
       instance_type: '${{parent.inputs.instance_type}}'
@@ -271,6 +282,7 @@ jobs:
       model_flavor: ${{parent.inputs.model_flavor}}
       vllm_enabled: ${{parent.inputs.vllm_enabled}}
       license_file_path: ${{parent.inputs.license_file_path}}
+      model_framework: ${{parent.inputs.model_framework}}
       model_download_metadata: ${{parent.jobs.download_model.outputs.model_download_metadata}}
       model_path: ${{parent.jobs.download_model.outputs.model_output}}
       hf_config_args: ${{parent.inputs.hf_config_args}}