upload flax model

allow flax

Update config.json

Update tokenizer_config.json

Initial commit

initial commit