Update blip.py

corajr · Jan 28, 2022 · b84fe14 · b84fe14
1 parent 0480d94
commit b84fe14
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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
-
+