How to use rajendrr/my-test-model with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModelForCausalLM tokenizer = AutoTokenizer.from_pretrained("rajendrr/my-test-model") model = AutoModelForCausalLM.from_pretrained("rajendrr/my-test-model", device_map="auto")
