code-summarization / download_model.py
trminhnam20082002's picture
feat: add repo
8b57e03
raw
history blame contribute delete
226 Bytes
import os
from huggingface_hub import hf_hub_download
path = hf_hub_download(
repo_id="tmnam20/codebert-code-summarization",
filename="pytorch_model.bin",
cache_dir="cache",
local_dir="models",
)
print(path)