diff --git a/models/blip.py b/models/blip.py index af28d59..e2813b6 100644 --- a/models/blip.py +++ b/models/blip.py @@ -5,6 +5,8 @@ * For full license text, see LICENSE.txt file in the repo root or https://opensource.org/licenses/BSD-3-Clause * By Junnan Li ''' +import warnings +warnings.filterwarnings("ignore") from models.vit import VisionTransformer, interpolate_pos_embed from models.med import BertConfig, BertModel, BertLMHeadModel @@ -233,4 +235,4 @@ def load_checkpoint(model,url_or_filename): msg = model.load_state_dict(state_dict,strict=False) print('load checkpoint from %s'%url_or_filename) return model,msg - \ No newline at end of file +