Load model directly

from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("tiiuae/falcon-40b") model = AutoModelForCausalLM.from_pretrained("tiiuae/falcon-40b")