import torch
from transformers import AutoModel
import os

# Hugging Face Hubからモデルをロード
# `safetensors`形式が自動的にダウンロードされ、PyTorchモデルとしてロードされます。
model = AutoModel.from_pretrained(
    "ku-nlp/deberta-v2-large-japanese-char-wwm",
    trust_remote_code=True
)

# 変換後のモデルの保存先パス
# `Dockerfile`のRUNコマンドで作成したディレクトリを指定します。
save_path = "bert/deberta-v2-large-japanese-char-wwm/pytorch_model.bin"

# モデルの重みをPyTorch形式で保存
torch.save(model.state_dict(), save_path)

print(f"Model successfully converted and saved to {save_path}")