blurr icon indicating copy to clipboard operation
blurr copied to clipboard

How to save and load the pretrained model to local?

Open nithinreddyy opened this issue 3 years ago • 0 comments

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)

nithinreddyy avatar Jul 13 '22 04:07 nithinreddyy