chore: fix imports in blip code due to file move

- models. 😭
gokayfem · Dec 24, 2022 · f081d1b · f081d1b
1 parent 82087b8
commit f081d1b
Show file tree

Hide file tree

Showing 6 changed files with 14 additions and 14 deletions.
diff --git a/blip/models/blip.py b/blip/models/blip.py
@@ -8,8 +8,8 @@
 import warnings
 warnings.filterwarnings("ignore")
 
-from models.vit import VisionTransformer, interpolate_pos_embed
-from models.med import BertConfig, BertModel, BertLMHeadModel
+from .vit import VisionTransformer, interpolate_pos_embed
+from .med import BertConfig, BertModel, BertLMHeadModel
 from transformers import BertTokenizer
 
 import torch

diff --git a/blip/models/blip_itm.py b/blip/models/blip_itm.py
@@ -1,11 +1,11 @@
-from models.med import BertConfig, BertModel
+from .med import BertConfig, BertModel
 from transformers import BertTokenizer
 
 import torch
 from torch import nn
 import torch.nn.functional as F
 
-from models.blip import create_vit, init_tokenizer, load_checkpoint
+from .blip import create_vit, init_tokenizer, load_checkpoint
 
 class BLIP_ITM(nn.Module):
     def __init__(self,                 

diff --git a/blip/models/blip_nlvr.py b/blip/models/blip_nlvr.py
@@ -1,7 +1,7 @@
-from models.med import BertConfig
-from models.nlvr_encoder import BertModel
-from models.vit import interpolate_pos_embed
-from models.blip import create_vit, init_tokenizer, is_url
+from .med import BertConfig
+from .nlvr_encoder import BertModel
+from .vit import interpolate_pos_embed
+from .blip import create_vit, init_tokenizer, is_url
 
 from timm.models.hub import download_cached_file
 

diff --git a/blip/models/blip_pretrain.py b/blip/models/blip_pretrain.py
@@ -5,7 +5,7 @@
  * For full license text, see LICENSE.txt file in the repo root or https://opensource.org/licenses/BSD-3-Clause
  * By Junnan Li
 '''
-from models.med import BertConfig, BertModel, BertLMHeadModel
+from .med import BertConfig, BertModel, BertLMHeadModel
 from transformers import BertTokenizer
 import transformers
 transformers.logging.set_verbosity_error()
@@ -14,7 +14,7 @@
 from torch import nn
 import torch.nn.functional as F
 
-from models.blip import create_vit, init_tokenizer, load_checkpoint
+from .blip import create_vit, init_tokenizer, load_checkpoint
 
 class BLIP_Pretrain(nn.Module):
     def __init__(self,                 

diff --git a/blip/models/blip_retrieval.py b/blip/models/blip_retrieval.py
@@ -1,11 +1,11 @@
-from models.med import BertConfig, BertModel
+from .med import BertConfig, BertModel
 from transformers import BertTokenizer
 
 import torch
 from torch import nn
 import torch.nn.functional as F
 
-from models.blip import create_vit, init_tokenizer, load_checkpoint
+from .blip import create_vit, init_tokenizer, load_checkpoint
 
 class BLIP_Retrieval(nn.Module):
     def __init__(self,                 

diff --git a/blip/models/blip_vqa.py b/blip/models/blip_vqa.py
@@ -1,5 +1,5 @@
-from models.med import BertConfig, BertModel, BertLMHeadModel
-from models.blip import create_vit, init_tokenizer, load_checkpoint
+from .med import BertConfig, BertModel, BertLMHeadModel
+from .blip import create_vit, init_tokenizer, load_checkpoint
 
 import torch
 from torch import nn