From b84fe149aef83a2943b7824962c583c70b28ab8c Mon Sep 17 00:00:00 2001 From: Junnan Li Date: Fri, 28 Jan 2022 23:09:56 +0800 Subject: [PATCH] Update blip.py --- models/blip.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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 +