How to use Pra-tham/mistral-training with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("Pra-tham/mistral-training", dtype="auto")