flax
support swap model params in gemma sampler
#4614
Merged

Loading