huggingface

下载模型

使用huggingface_hub api可以下载社区的模型
pip install huggingface_hub

from huggingface_hub import snapshot_download
snapshot_download(repo_id="google/vit-base-patch16-224", allow_patterns=["*.json", "pytorch_model.bin", "vocab.txt"], local_dir="./my_model/")

加载模型使用

VIT_MODEL_NAME_OR_PATH = "google/vit-base-patch16-224"
vision_encoder_decoder_model_name_or_path='./my_model/'
model = VisionEncoderDecoderModel.from_pretrained(vision_encoder_decoder_model_name_or_path)

参考自csdn