blurr
blurr copied to clipboard
How to save and load the pretrained model to local?
I'm working on the text summarization task. How to save and load the pretrained model instead of downloading again and again on google colab/local pc?
Below is the code i'm working on
pretrained_model_name = "facebook/bart-large-cnn"
hf_arch, hf_config, hf_tokenizer, hf_model = get_hf_objects(pretrained_model_name, model_cls=BartForConditionalGeneration)
hf_arch, type(hf_config), type(hf_tokenizer), type(hf_model)