upload flax model

allow flax

added missing tokenizers

added tokenizer in the config file

CHekcing in the checpoints

Modified config.json

Update from Bala

initial commit