[CoreEngine] update the workflow example.

shatrugna · Apr 30, 2024 · 2fbb96d · 2fbb96d
1 parent 21cc0e3
commit 2fbb96d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/fedml/workflow/driver_example/customized_job_example/customized_workflow.py b/python/fedml/workflow/driver_example/customized_job_example/customized_workflow.py
@@ -171,7  171,7 @@ def create_train_workflow(job_api_key=None):
 
     if is_inference and deployed_endpoint_name is not None:
         create_inference_workflow(
-            job_api_key=args.api_key, endpoint_name_list=[deployed_endpoint_name, "fedml-dimitris-mistral-7b"],
             job_api_key=args.api_key, endpoint_name_list=[deployed_endpoint_name],
             input_json=args.infer_json, user_name=args.user_name)
 
     if is_train: