forward method of
GPT2Model allows the user to pass the embedded representation of tokens directly to GPT2 using the
inputs_embeds keyword. However I notice that the
__call__ method of
FlaxGPT2Model does not seem to accept this keyword. Is there a workaround to achieve the same functionality? Thanks.