transformers safetensors mistral text-generation conversational en dataset:cais/wmdp dataset:cais/wmdp-corpora arxiv:2403.03218 license:mit autotrain_compatible text-generation-inference endpoints_compatible region:us