from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("meta-llama/Meta-Llama-3.1-405B")
model = AutoModelForCausalLM.from_pretrained("meta-llama/Meta-Llama-3.1-405B")