transformers safetensors mistral text-generation conversational dataset:Azma-AI/azma-final-cot-agent-dataset arxiv:1910.09700 autotrain_compatible text-generation-inference endpoints_compatible region:us