Skip to content

Commit

Permalink
fix missing job variable
Browse files Browse the repository at this point in the history
  • Loading branch information
Benjoyo committed Apr 30, 2024
1 parent d2493ae commit 2b9b949
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion bpm-ai-connectors-c8/bpm_ai_connectors_c8/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down

0 comments on commit 2b9b949

Please sign in to comment.