[Flax generate] Add params to generate (#12171)
[Flax generate] Add params to generate (#12171)

* fix_torch_device_generate_test

* remove @

* add params as input

* finish
huggingface/transformers,huggingface/transformers,huggingface/transformers,huggingface/transformers