model_name2path = {
                    'gpt2-large':'gpt2-large', # path to your gpt2-large checkpoint
                    'gpt2-medium':'gpt2-medium', # path to your gpt2-medium checkpoint
                    'llama-7b': 'llama-7b', # path to your Llama-2 checkpoint
                    'gpt2':'gpt2', # path to your gpt2 checkpoint
                    'gpt2-xl':'gpt2-xl', # path to your gpt2-xl checkpoint
                }