transformers safetensors mistral text-generation generated_from_trainer dataset:nilq/baby-python arxiv:2410.12391 model-index autotrain_compatible text-generation-inference endpoints_compatible region:us