error while loading RetinaFace .h5

[luikaire]

On running
python test_image_swap_multi.py --facedancer_path "./model_zoo/FaceDancer_config_c_HQ.h5" --img_path "data/in1.jpg" --swap_source "data/in2.jpg" --img_output "results/swapped_image.jpg"

Initializing FaceDancer...
Traceback (most recent call last):
  File "/home/ani/code/misc/deepfake/FaceDancer/test_image_swap_multi.py", line 28, in <module>
    RetinaFace = load_model(opt.retina_path, compile=False,
  File "/home/ani/.local/lib/python3.10/site-packages/keras/src/saving/saving_api.py", line 183, in load_model
    return legacy_h5_format.load_model_from_hdf5(filepath)
  File "/home/ani/.local/lib/python3.10/site-packages/keras/src/legacy/saving/legacy_h5_format.py", line 133, in load_model_from_hdf5
    model = saving_utils.model_from_config(
  File "/home/ani/.local/lib/python3.10/site-packages/keras/src/legacy/saving/saving_utils.py", line 83, in model_from_config
    config = _find_replace_nested_dict(config, "keras.", "keras.")
  File "/home/ani/.local/lib/python3.10/site-packages/keras/src/legacy/saving/saving_utils.py", line 230, in _find_replace_nested_dict
    dict_str = json.dumps(config)
  File "/usr/lib/python3.10/json/__init__.py", line 231, in dumps
    return _default_encoder.encode(obj)
  File "/usr/lib/python3.10/json/encoder.py", line 199, in encode
    chunks = self.iterencode(o, _one_shot=True)
  File "/usr/lib/python3.10/json/encoder.py", line 257, in iterencode
    return _iterencode(o, 0)
  File "/usr/lib/python3.10/json/encoder.py", line 179, in default
    raise TypeError(f'Object of type {o.__class__.__name__} '
TypeError: Object of type ellipsis is not JSON serializable

I'm using tensorflow-nightly, keras-nightly, tfa-nightly