File size: 568 Bytes
a5b6d87 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
from transformers import BlipProcessor, BlipModel
# Load model & processor
processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
model = BlipModel.from_pretrained("Salesforce/blip-image-captioning-base")
from huggingface_hub import notebook_login
notebook_login()
processor.save_pretrained('blip-base') # safe_serialization=True
model.save_pretrained('blip-base')
!hf repo create blip-base --repo-type model
from huggingface_hub import upload_folder
upload_folder(
repo_id='DevJGraham/blip-base',
folder_path = 'blip-base'
) |