Adding `safetensors` variant of this model

upload flax model

allow flax

Update model preview

Update model preview

New model from https://wandb.ai/wandb/huggingtweets/runs/3tlt3nyy

initial commit