diff --git a/bpm-ai-connectors-c8/bpm_ai_connectors_c8/models.py b/bpm-ai-connectors-c8/bpm_ai_connectors_c8/models.py index 1a6f8bf..d9dd100 100644 --- a/bpm-ai-connectors-c8/bpm_ai_connectors_c8/models.py +++ b/bpm-ai-connectors-c8/bpm_ai_connectors_c8/models.py @@ -21,7 +21,7 @@ # job variable names that define model parameters that should not be injected into the task function # (instead only the final constructed model object is injected) -EXTRA_VARS = ["custom_asr", "custom_classifier", "custom_image_classifier", "custom_qa", "custom_vqa", "custom_nmt", "model_endpoint", "model_filename"] +EXTRA_VARS = ["custom_asr", "custom_classifier", "custom_image_classifier", "custom_qa", "custom_vqa", "custom_nmt", "model_endpoint", "local_llm", "model_filename"] def model_ids_to_models(kwargs: dict):