| 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) |