How to use Badribn/mistral_prompt_tuning with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("Badribn/mistral_prompt_tuning", device_map="auto")
