Flax-community/koclip error

I followed the below instructions, but I still got the same error.

This is colab link. I get the same error

TypeError: __init__() got an unexpected keyword argument '_do_init'
Traceback:
File "/home/user/.local/lib/python3.8/site-packages/streamlit/scriptrunner/script_runner.py", line 554, in _run_script
    exec(code, module.__dict__)
File "/home/user/app/app.py", line 20, in <module>
    PAGES[page].app(model)
File "/home/user/app/text2image.py", line 15, in app
    model, processor = load_model(f"koclip/{model_name}")
File "/home/user/app/utils.py", line 49, in load_model
    cached_model = load_model_cached(model_name)
File "/home/user/.local/lib/python3.8/site-packages/streamlit/legacy_caching/caching.py", line 573, in wrapped_func
    return get_or_create_cached_value()
File "/home/user/.local/lib/python3.8/site-packages/streamlit/legacy_caching/caching.py", line 557, in get_or_create_cached_value
    return_value = func(*args, **kwargs)
File "/home/user/app/utils.py", line 58, in load_model_cached
    model = FlaxHybridCLIP.from_pretrained(model_name)
File "/home/user/.local/lib/python3.8/site-packages/transformers/modeling_flax_utils.py", line 784, in from_pretrained
    model = cls(config, *model_args, _do_init=_do_init, **model_kwargs)
File "/home/user/app/koclip/model.py", line 161, in __init__
    module = self.module_class(config=config, dtype=dtype, **kwargs)