Equivalent of `inputs_embeds` for `FlaxGPT2Model`
Hi,Theforwardmethod ofGPT2Modelallows the user to pass the embedded representation of tokens directly to GPT2 using theinputs_embedskeyword. However I notice that the__call__method ofFlaxGPT2Modeldoes not seem to accept this keyword. Is there a workaround to achieve the same functionality? Thanks.