diff --git a/src/evaluate_params.py b/src/evaluate_params.py index 1619cf6ab..d65ac0ef0 100644 --- a/src/evaluate_params.py +++ b/src/evaluate_params.py @@ -83,6 +83,7 @@ 'image_file', 'image_control', + 'base_model' ] # form evaluate defaults for submit_nochat_api diff --git a/src/gen.py b/src/gen.py index c35a6f9dc..484f8bcd5 100644 --- a/src/gen.py +++ b/src/gen.py @@ -4529,7 +4529,7 @@ def evaluate( if not stream_output and img_file == 1: from src.vision.utils_vision import get_llava_response response, _ = get_llava_response(**llava_kwargs) - + yield dict(response=response, sources=[], save_dict={}, error='', llm_answers={}, response_no_refs=response, sources_str='', prompt_raw='') else: @@ -4665,6 +4665,7 @@ def evaluate( image_file=img_file, image_control=None, # already stuffed into image_file + base_model=base_model, ) assert len(set(list(client_kwargs.keys())).symmetric_difference(eval_func_param_names)) == 0 api_name = '/submit_nochat_api' # NOTE: like submit_nochat but stable API for string dict passing @@ -5396,6 +5397,7 @@ def mean(a):""", ''] + params_list, tts_speed, image_file, image_control, + None, ] # adjust examples if non-chat mode if not chat: