from transformers import AutoModelForCausalLM, AutoTokenizer

# モデル名を指定
model_name = "./"

# モデルとトークナイザーを読み込む
model = AutoModelForCausalLM.from_pretrained(model_name)
# tokenizer = AutoTokenizer.from_pretrained(model_name)

# モデルの状態ディクショナリを取得
state_dict = model.state_dict()

# weightキーを表示
for key in state_dict.keys():
    if "weight" in key:
        print(key)