1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_id = "YOUSSEF88/generative-dim-collapse-paper"
4tokenizer = AutoTokenizer.from_pretrained(model_id)
5model = AutoModelForCausalLM.from_pretrained(model_id)AutoModelForCausalLM with the appropriate AutoModel class.