keras 3 backward compatibility

Author: ChaoPang

src/MEDS_DEV/models/cehrxgpt/mimiciv/model.yaml

@@ -15,6 +15,8 @@ commands:
       mkdir -p "{output_dir}/cehrgpt_pretrained/dataset_prepared"
       meds_reader_convert "{dataset_dir}" "{output_dir}/cehrgpt_pretrained/meds_reader" --num_threads 8
 
+      pip install tf-keras
+
       export CUDA_VISIBLE_DEVICES="0"; python -u -m cehrgpt.runners.hf_cehrgpt_pretrain_runner \
       --model_name_or_path "{output_dir}/cehrgpt_pretrained" \
       --tokenizer_name_or_path "{output_dir}/cehrgpt_pretrained" \

src/MEDS_DEV/models/cehrxgpt/omop/model.yaml

@@ -15,6 +15,8 @@ commands:
       mkdir -p "{output_dir}/cehrgpt_pretrained/dataset_prepared"
       meds_reader_convert "{dataset_dir}" "{output_dir}/cehrgpt_pretrained/meds_reader" --num_threads 8
 
+      pip install tf-keras
+
       export CUDA_VISIBLE_DEVICES="0"; python -u -m cehrgpt.runners.hf_cehrgpt_pretrain_runner \
       --model_name_or_path "{output_dir}/cehrgpt_pretrained" \
       --tokenizer_name_or_path "{output_dir}/cehrgpt_pretrained" \